From 29e27b0544d99e5d98f044a9e606db8abcfb8900 Mon Sep 17 00:00:00 2001 From: Pablo Marquez Tello Date: Thu, 3 Aug 2023 14:47:31 +0100 Subject: Add support for S64 output in NEArgMinMaxLayer * NEArgMinMaxLayer uses NEReductionOperation to compute its result in S32 * We need to call NECast to convert from S32 to S64 * Resolves MLCE-1089 Change-Id: I6fded869b6076d7af1b9b3e70eb384f4ee82fd8a Signed-off-by: Pablo Marquez Tello Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10054 Tested-by: Arm Jenkins Reviewed-by: Viet-Hoa Do Comments-Addressed: Arm Jenkins Benchmark: Arm Jenkins --- docs/user_guide/operator_list.dox | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'docs/user_guide/operator_list.dox') diff --git a/docs/user_guide/operator_list.dox b/docs/user_guide/operator_list.dox index 66b8988d29..e0b45419eb 100644 --- a/docs/user_guide/operator_list.dox +++ b/docs/user_guide/operator_list.dox @@ -126,7 +126,7 @@ where N = batches, C = channels, H = height, W = width, D = depth srcdst QASYMM8U32, S32 QASYMM8_SIGNEDU32, S32 - S32U32, S32 + S32U32, S32, S64 F16U32, S32 F32U32, S32 -- cgit v1.2.1