fix reduce param set

5 years ago · 0fd28fd435
parent 8946acc35b
commit 0fd28fd435
1 changed files with 2 additions and 1 deletions
--- 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;