From 0fd28fd4355883caf7f9cbb714b5a791b292acc6 Mon Sep 17 00:00:00 2001
From: zhaozhenlong <zhaozhenlong1@huawei.com>
Date: Fri, 18 Sep 2020 14:58:13 +0800
Subject: [PATCH] fix reduce param set

---
 mindspore/lite/src/ops/reduce.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mindspore/lite/src/ops/reduce.cc b/mindspore/lite/src/ops/reduce.cc
index ad6635522c..2fd893f405 100644
--- a/mindspore/lite/src/ops/reduce.cc
+++ b/mindspore/lite/src/ops/reduce.cc
@@ -118,7 +118,8 @@ int Reduce::UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers:
       axes.push_back(attr->axes()->data()[i]);
     }
   }
-  auto val_offset = schema::CreateReduceDirect(*fbb, &axes, attr->keepDims(), attr->mode());
+  auto val_offset =
+    schema::CreateReduceDirect(*fbb, &axes, attr->keepDims(), attr->mode(), attr->reduceToEnd(), attr->coeff());
   auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_Reduce, val_offset.o);
   fbb->Finish(prim_offset);
   return RET_OK;