From 4ebdccad11b66d2603db440bd379b1eb00d084c5 Mon Sep 17 00:00:00 2001
From: cjh9368 <chengjiahui@huawei.com>
Date: Wed, 27 Jan 2021 17:02:00 +0800
Subject: [PATCH] fix caffe reduce parser bug

---
 .../lite/tools/converter/parser/caffe/caffe_reduce_parser.cc     | 1 +
 1 file changed, 1 insertion(+)

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<int>(1, 0);
   }
   attr->axes = axes;
+  attr->reduceToEnd = true;
 
   auto primitive = std::make_unique<schema::PrimitiveT>();
   primitive->value.type = schema::PrimitiveType_Reduce;