diff options
author | Pablo Marquez Tello <pablo.tello@arm.com> | 2023-08-03 14:47:31 +0100 |
---|---|---|
committer | Pablo Marquez Tello <pablo.tello@arm.com> | 2023-08-08 15:49:54 +0000 |
commit | 29e27b0544d99e5d98f044a9e606db8abcfb8900 (patch) | |
tree | 3749d3f3640d55fceda4dcd04a2916c87414b045 /docs/user_guide/operator_list.dox | |
parent | 66b4a6a8ca1ee55e5b7f05bae2543cf99fe22d6d (diff) | |
download | ComputeLibrary-29e27b0544d99e5d98f044a9e606db8abcfb8900.tar.gz |
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>
Diffstat (limited to 'docs/user_guide/operator_list.dox')
-rw-r--r-- | docs/user_guide/operator_list.dox | 2 |
1 files changed, 1 insertions, 1 deletions
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 <tr><th>src<th>dst <tr><td>QASYMM8<td>U32, S32 <tr><td>QASYMM8_SIGNED<td>U32, S32 - <tr><td>S32<td>U32, S32 + <tr><td>S32<td>U32, S32, S64 <tr><td>F16<td>U32, S32 <tr><td>F32<td>U32, S32 </table> |