!4012 fix squeeze op bug

Merge pull request !4012 from mengchunyang/master

mindspore/lite/src/model_impl.cc

@@ -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:

mindspore/lite/src/populate_parameter.cc

@@ -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;