!13794 [MS][LITE][CPU]fix bug of reduce

From: @fuzhiye
Reviewed-by: @zhanghaibo5,@zhang_xue_tong
Signed-off-by: @zhang_xue_tong
pull/13794/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit fb69f3e509

@ -125,7 +125,7 @@ int ReduceCPUKernel::Run() {
}
src_data_ = dst_data_;
}
if (reduce_param_->reduce_to_end_ && abs(reduce_param_->coeff - 1.0f) > 1e-5) {
if (reduce_param_->reduce_to_end_ && abs(reduce_param_->coeff) > 1e-5) {
ret = CalculateCoeffOutput();
if (ret != RET_OK) {
FreeTmpBuffer();

Loading…
Cancel
Save