diff options
author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-10-12 21:44:23 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-10-13 12:11:51 +0000 |
commit | cbede286da8711031fb6fc56bb2e2c246a4c5455 (patch) | |
tree | a952431d502aa0f5c19ae4008323b5bc6cc2d127 /src/core/NEON | |
parent | 141c31a532efb20698466aaabbecc92639f05b0d (diff) | |
download | ComputeLibrary-cbede286da8711031fb6fc56bb2e2c246a4c5455.tar.gz |
COMPMID-3828: fix unsigned overflow in quantized reduce mean
Change-Id: I9d3122b4858137d422548d1d417eb04a27ae9c7b
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4143
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: TeresaARM <teresa.charlinreyes@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON')
-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; } |