diff --git a/mindspore/lite/tools/converter/parser/caffe/caffe_flatten_parser.cc b/mindspore/lite/tools/converter/parser/caffe/caffe_flatten_parser.cc index 7f78c42b98cccaa20578c517fdbc787bd484e222..2e17f6ddd36d89a7d9c1d54671acece1a4b2fc1e 100644 --- a/mindspore/lite/tools/converter/parser/caffe/caffe_flatten_parser.cc +++ b/mindspore/lite/tools/converter/parser/caffe/caffe_flatten_parser.cc @@ -24,13 +24,7 @@ STATUS CaffeFlattenParser::Parse(const caffe::LayerParameter &proto, const caffe // MS_LOG(ERROR) << "null pointer dereferencing."; return RET_NULL_PTR; } - std::unique_ptr attr(new schema::FullConnectionT()); - const caffe::FlattenParameter flattenParam = proto.flatten_param(); - - attr->axis = (int32_t)flattenParam.axis(); - attr->useAxis = true; - attr->hasBias = false; - attr->activationType = schema::ActivationType_NO_ACTIVATION; + std::unique_ptr attr = std::make_unique(); op->primitive = std::make_unique(); op->primitive->value.type = schema::PrimitiveType_Flatten; diff --git a/mindspore/lite/tools/converter/parser/caffe/caffe_innerproduct_parser.cc b/mindspore/lite/tools/converter/parser/caffe/caffe_innerproduct_parser.cc index 5667132b40b8766f6bfb5218f5dd5b37bbee4c42..1cc01dcedcdf0b175355253e69587a08f2864eb4 100644 --- a/mindspore/lite/tools/converter/parser/caffe/caffe_innerproduct_parser.cc +++ b/mindspore/lite/tools/converter/parser/caffe/caffe_innerproduct_parser.cc @@ -40,6 +40,7 @@ STATUS CaffeInnerProductParser::Parse(const caffe::LayerParameter &proto, const if (innerProductParam.bias_term()) { attr->hasBias = true; } + attr->activationType = schema::ActivationType_NO_ACTIVATION; // parse weight if (weight.blobs_size() == 0) { diff --git a/mindspore/lite/tools/converter/parser/onnx/onnx_deconv_parser.cc b/mindspore/lite/tools/converter/parser/onnx/onnx_deconv_parser.cc index c0ff092695e3eedf732dd4f19375b21fcb07ede2..09236775fe2620faabadf99b758da90a9acbe3eb 100644 --- a/mindspore/lite/tools/converter/parser/onnx/onnx_deconv_parser.cc +++ b/mindspore/lite/tools/converter/parser/onnx/onnx_deconv_parser.cc @@ -49,7 +49,7 @@ bool OnnxDeConvParser::ParseGroupDeConvolution(schema::CNodeT *op, schema::DeCon deDepthwiseConv2DParam->activationType = attr->activationType; if (op != nullptr) { op->primitive = std::make_unique(); - op->primitive->value.type = schema::PrimitiveType_DepthwiseConv2D; + op->primitive->value.type = schema::PrimitiveType_DeDepthwiseConv2D; delete (op->primitive->value.value); op->primitive->value.value = deDepthwiseConv2DParam; }