aboutsummaryrefslogtreecommitdiff
path: root/tests/SimpleTensor.h
diff options
context:
space:
mode:
authorPablo Marquez Tello <pablo.tello@arm.com>2023-07-18 14:51:24 +0100
committerPablo Marquez Tello <pablo.tello@arm.com>2023-07-19 15:21:03 +0000
commit4a1c91767142f76e92bf4575564d7e54fcd0ebf4 (patch)
tree9ff60356ad77b07c3704826c14eb316a9eb9ac27 /tests/SimpleTensor.h
parent314d3e2c691734ff942c9a1aca1120b826a267d9 (diff)
downloadComputeLibrary-4a1c91767142f76e92bf4575564d7e54fcd0ebf4.tar.gz
Add support for input S64/U64 in CpuCastKernel
* The kernel now supports the following conversions: S64 -> F32 U64 -> F32 * Resolves MLCE-1089 Change-Id: I277cf58b78d919fde25947520d2056e1412c7f82 Signed-off-by: Pablo Marquez Tello <pablo.tello@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9935 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 'tests/SimpleTensor.h')
-rw-r--r--tests/SimpleTensor.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/SimpleTensor.h b/tests/SimpleTensor.h
index 9ea171d492..419621e808 100644
--- a/tests/SimpleTensor.h
+++ b/tests/SimpleTensor.h
@@ -392,6 +392,8 @@ int SimpleTensor<T>::num_channels() const
case Format::S16:
case Format::U32:
case Format::S32:
+ case Format::U64:
+ case Format::S64:
case Format::F16:
case Format::F32:
return 1;