|
|
|
@@ -40,6 +40,7 @@ int TransposeFp16CPUKernel::Init() { |
|
|
|
int TransposeFp16CPUKernel::Run() { |
|
|
|
MS_ASSERT(in_tensors_.size() == 1 || in_tensors_.size() == 2); |
|
|
|
TransposeParameter *param = reinterpret_cast<TransposeParameter *>(this->op_parameter_); |
|
|
|
param->data_size_ = in_tensors_[0]->Size(); |
|
|
|
if (in_tensors_.size() == 2) { |
|
|
|
auto input_perm = in_tensors_.at(1); |
|
|
|
MS_ASSERT(input_perm != nullptr); |
|
|
|
|