From eaa01ab593428bc7267ebbe107b2d813a11b64b5 Mon Sep 17 00:00:00 2001 From: Sang-Hoon Park Date: Mon, 11 Nov 2019 17:33:28 +0000 Subject: COMPMID-2671 use Signed32 for default output data type of ArgMinMax Signed32 is used as data types before and after reshaping of ArgMinMax. Change-Id: I230af43a931d4e106de6c72f716ced1dab511084 Signed-off-by: Sang-Hoon Park Reviewed-on: https://review.mlplatform.org/c/2262 Comments-Addressed: Arm Jenkins Reviewed-by: Michele Di Giorgio Tested-by: Arm Jenkins Reviewed-by: Giorgio Arena --- arm_compute/core/NEON/kernels/NEReductionOperationKernel.h | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'arm_compute/core/NEON/kernels/NEReductionOperationKernel.h') diff --git a/arm_compute/core/NEON/kernels/NEReductionOperationKernel.h b/arm_compute/core/NEON/kernels/NEReductionOperationKernel.h index 4b28b8dbcd..0d402fb044 100644 --- a/arm_compute/core/NEON/kernels/NEReductionOperationKernel.h +++ b/arm_compute/core/NEON/kernels/NEReductionOperationKernel.h @@ -32,10 +32,10 @@ class ITensor; /** NEON kernel to perform a reduction operation * - * @note For ARG_MIN/ARG_MAX reduction, the indices are computed in unsigned - * 32-bit (U32). It is the user's responsibility to check that the - * results do not overflow in case the output data type is set to signed - * 32-bit integer (S32). + * @note For ARG_MIN/ARG_MAX reduction, the default data type for an uninitialized + * output tensor is signed 32-bit integer (S32). It is the user's responsibility + * to check that the results do not overflow because the indices are computed + * in unsigned 32-bit (U32). */ class NEReductionOperationKernel : public INEKernel { -- cgit v1.2.1