!7257 add op_parameter for nchw2nhwc and nhwc2nchw

Merge pull request !7257 from hangq/master
pull/7257/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 6999cf9c55

@ -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) {

Loading…
Cancel
Save