Browse Source

fix bug of squeeze op

tags/v0.7.0-beta
meng_chunyang 5 years ago
parent
commit
49e94a8559
2 changed files with 3 additions and 1 deletions
  1. +2
    -0
      mindspore/lite/src/model_impl.cc
  2. +1
    -1
      mindspore/lite/src/populate_parameter.cc

+ 2
- 0
mindspore/lite/src/model_impl.cc View File

@@ -132,6 +132,8 @@ lite::Primitive *ModelImpl::CopyPrimitive(const schema::Primitive *srcPrim) {
return new lite::Transpose(const_cast<schema::Primitive *>(srcPrim));
case schema::PrimitiveType_Slice:
return new lite::Slice(const_cast<schema::Primitive *>(srcPrim));
case schema::PrimitiveType_Squeeze:
return new lite::Squeeze(const_cast<schema::Primitive *>(srcPrim));
case schema::PrimitiveType_Nchw2Nhwc:
return new lite::Nchw2Nhwc(const_cast<schema::Primitive *>(srcPrim));
case schema::PrimitiveType_Nhwc2Nchw:


+ 1
- 1
mindspore/lite/src/populate_parameter.cc View File

@@ -1222,7 +1222,7 @@ PopulateParameterRegistry::PopulateParameterRegistry() {
populate_parameter_funcs_[schema::PrimitiveType_AddN] = PopulateAddNParameter;
populate_parameter_funcs_[schema::PrimitiveType_StridedSlice] = PopulateStridedSliceParameter;
populate_parameter_funcs_[schema::PrimitiveType_ScatterND] = PopulateScatterNDParameter;
populate_parameter_funcs_[schema::PrimitiveType_Square] = PopulateSqueezeParameter;
populate_parameter_funcs_[schema::PrimitiveType_Squeeze] = PopulateSqueezeParameter;
populate_parameter_funcs_[schema::PrimitiveType_Split] = PopulateSplitParameter;
populate_parameter_funcs_[schema::PrimitiveType_PriorBox] = PopulatePriorBoxParameter;
populate_parameter_funcs_[schema::PrimitiveType_QuantDTypeCast] = PopulateQuantDTypeCastParameter;


Loading…
Cancel
Save