tree 3749d3f3640d55fceda4dcd04a2916c87414b045
parent 66b4a6a8ca1ee55e5b7f05bae2543cf99fe22d6d
author Pablo Marquez Tello <pablo.tello@arm.com> 1691070451 +0100
committer Pablo Marquez Tello <pablo.tello@arm.com> 1691509794 +0000

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 <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10054
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>