| @@ -863,25 +863,35 @@ OpParameter *PopulateTopKParameter(const mindspore::lite::PrimitiveC *primitive) | |||||
| } | } | ||||
| OpParameter *PopulateNhwc2NchwParameter(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) { | if (parameter == nullptr) { | ||||
| MS_LOG(ERROR) << "malloc OpParameter failed."; | MS_LOG(ERROR) << "malloc OpParameter failed."; | ||||
| return nullptr; | return nullptr; | ||||
| } | } | ||||
| memset(parameter, 0, sizeof(OpParameter)); | 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 *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) { | if (parameter == nullptr) { | ||||
| MS_LOG(ERROR) << "malloc OpParameter failed."; | MS_LOG(ERROR) << "malloc OpParameter failed."; | ||||
| return nullptr; | return nullptr; | ||||
| } | } | ||||
| memset(parameter, 0, sizeof(OpParameter)); | 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) { | OpParameter *PopulateTransposeParameter(const mindspore::lite::PrimitiveC *primitive) { | ||||