Browse Source

!6476 fix reduce parameter transfer bug

Merge pull request !6476 from zhaozhenlong/lite/issue/reduce_coefficient
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
94f9ddadfe
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/lite/src/ops/reduce.cc

+ 2
- 1
mindspore/lite/src/ops/reduce.cc View File

@@ -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;


Loading…
Cancel
Save