Browse Source

fix reduce param set

tags/v1.0.0
zhaozhenlong 5 years ago
parent
commit
0fd28fd435
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