diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc index d0f61a8098..c221536bfa 100644 --- 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();