fix bug of reduce

4 years ago · 287988553e
parent ab5d77f943
commit 287988553e
1 changed files with 1 additions and 1 deletions
--- a/mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc
@ -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();