|
|
|
@@ -863,25 +863,35 @@ OpParameter *PopulateTopKParameter(const mindspore::lite::PrimitiveC *primitive) |
|
|
|
} |
|
|
|
|
|
|
|
OpParameter *PopulateNhwc2NchwParameter(const mindspore::lite::PrimitiveC *primitive) { |
|
|
|
OpParameter *parameter = reinterpret_cast<OpParameter *>(malloc(sizeof(OpParameter))); |
|
|
|
TransposeParameter *parameter = reinterpret_cast<TransposeParameter *>(malloc(sizeof(TransposeParameter))); |
|
|
|
if (parameter == nullptr) { |
|
|
|
MS_LOG(ERROR) << "malloc OpParameter failed."; |
|
|
|
return nullptr; |
|
|
|
} |
|
|
|
memset(parameter, 0, sizeof(OpParameter)); |
|
|
|
parameter->type_ = primitive->Type(); |
|
|
|
return parameter; |
|
|
|
parameter->op_parameter_.type_ = primitive->Type(); |
|
|
|
parameter->num_axes_ = 4; |
|
|
|
parameter->perm_[0] = 0; |
|
|
|
parameter->perm_[1] = 3; |
|
|
|
parameter->perm_[2] = 1; |
|
|
|
parameter->perm_[3] = 2; |
|
|
|
return reinterpret_cast<OpParameter *>(parameter); |
|
|
|
} |
|
|
|
|
|
|
|
OpParameter *PopulateNchw2NhwcParameter(const mindspore::lite::PrimitiveC *primitive) { |
|
|
|
OpParameter *parameter = reinterpret_cast<OpParameter *>(malloc(sizeof(OpParameter))); |
|
|
|
TransposeParameter *parameter = reinterpret_cast<TransposeParameter *>(malloc(sizeof(TransposeParameter))); |
|
|
|
if (parameter == nullptr) { |
|
|
|
MS_LOG(ERROR) << "malloc OpParameter failed."; |
|
|
|
return nullptr; |
|
|
|
} |
|
|
|
memset(parameter, 0, sizeof(OpParameter)); |
|
|
|
parameter->type_ = primitive->Type(); |
|
|
|
return parameter; |
|
|
|
parameter->op_parameter_.type_ = primitive->Type(); |
|
|
|
parameter->num_axes_ = 4; |
|
|
|
parameter->perm_[0] = 0; |
|
|
|
parameter->perm_[1] = 2; |
|
|
|
parameter->perm_[2] = 3; |
|
|
|
parameter->perm_[3] = 1; |
|
|
|
return reinterpret_cast<OpParameter *>(parameter); |
|
|
|
} |
|
|
|
|
|
|
|
OpParameter *PopulateTransposeParameter(const mindspore::lite::PrimitiveC *primitive) { |
|
|
|
|