diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/core/NEON/kernels/NEReductionOperationKernel.cpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/NEReductionOperationKernel.cpp b/src/core/NEON/kernels/NEReductionOperationKernel.cpp index 9b8c971a35..afe58ed07d 100644 --- a/src/core/NEON/kernels/NEReductionOperationKernel.cpp +++ b/src/core/NEON/kernels/NEReductionOperationKernel.cpp @@ -724,14 +724,13 @@ struct RedOpX_quantized if(op == ReductionOperation::MEAN_SUM) { - res /= in_info.dimension(0); + res /= static_cast<int32_t>(in_info.dimension(0)); } else { // Subtract accumulated offsets res -= (in_info.dimension(0) - 1) * iq_info.offset; } - *reinterpret_cast<T *>(output.ptr()) = utils::cast::saturate_cast<T>(res); } } |