From 4a1c91767142f76e92bf4575564d7e54fcd0ebf4 Mon Sep 17 00:00:00 2001 From: Pablo Marquez Tello Date: Tue, 18 Jul 2023 14:51:24 +0100 Subject: 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 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9935 Tested-by: Arm Jenkins Reviewed-by: Viet-Hoa Do Comments-Addressed: Arm Jenkins Benchmark: Arm Jenkins --- tests/SimpleTensor.h | 2 ++ 1 file changed, 2 insertions(+) (limited to 'tests/SimpleTensor.h') 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::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; -- cgit v1.2.1