diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/core/NEON/kernels/NEReductionOperationKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/NEON/kernels/NEReductionOperationKernel.cpp b/src/core/NEON/kernels/NEReductionOperationKernel.cpp index 9eebb42151..9af7f2ab10 100644 --- a/src/core/NEON/kernels/NEReductionOperationKernel.cpp +++ b/src/core/NEON/kernels/NEReductionOperationKernel.cpp @@ -1459,7 +1459,7 @@ struct RedOpYZW_quantized case ReductionOperation::MEAN_SUM: { int32_t res = static_cast<int32_t>(res_value); - res /= in_info.dimension(axis); + res /= static_cast<int32_t>(in_info.dimension(axis)); *reinterpret_cast<T *>(output.ptr() + x) = utils::cast::saturate_cast<T>(res); break; } |