Browse Source

fix bug of reduce

pull/13794/head
fuzhiye 5 years ago
parent
commit
287988553e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc

+ 1
- 1
mindspore/lite/src/runtime/kernel/arm/fp32/reduce_fp32.cc View File

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