diff --git a/mindspore/lite/tools/converter/parser/caffe/caffe_reduce_parser.cc b/mindspore/lite/tools/converter/parser/caffe/caffe_reduce_parser.cc index 0961e6fb40..f13bc206fe 100644 --- a/mindspore/lite/tools/converter/parser/caffe/caffe_reduce_parser.cc +++ b/mindspore/lite/tools/converter/parser/caffe/caffe_reduce_parser.cc @@ -55,6 +55,7 @@ PrimitiveC *CaffeReduceParser::ParseLitePrimitive(const caffe::LayerParameter &p axes = std::vector(1, 0); } attr->axes = axes; + attr->reduceToEnd = true; auto primitive = std::make_unique(); primitive->value.type = schema::PrimitiveType_Reduce;