diff options
Diffstat (limited to 'arm_compute/runtime')
-rw-r--r-- | arm_compute/runtime/CL/functions/CLArgMinMaxLayer.h | 7 | ||||
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h | 7 |
2 files changed, 8 insertions, 6 deletions
diff --git a/arm_compute/runtime/CL/functions/CLArgMinMaxLayer.h b/arm_compute/runtime/CL/functions/CLArgMinMaxLayer.h index 28feee09ab..1b465a4866 100644 --- a/arm_compute/runtime/CL/functions/CLArgMinMaxLayer.h +++ b/arm_compute/runtime/CL/functions/CLArgMinMaxLayer.h @@ -38,9 +38,10 @@ class CLReductionOperation; /** Function to calculate the index of the minimum or maximum values in a * tensor based on an axis. * - * @note 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 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 CLArgMinMaxLayer : public IFunction { diff --git a/arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h b/arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h index d2ddff2665..ca6794bf82 100644 --- a/arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h +++ b/arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h @@ -42,9 +42,10 @@ class ITensor; * -# @ref NEReductionOperationKernel * -# @ref NEFillBorderKernel * - * @note 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 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 NEArgMinMaxLayer : public IFunction { |