diff options
| author | Pablo Marquez Tello <pablo.tello@arm.com> | 2023-07-18 14:51:24 +0100 |
|---|---|---|
| committer | Pablo Marquez Tello <pablo.tello@arm.com> | 2023-07-19 15:21:03 +0000 |
| commit | 4a1c91767142f76e92bf4575564d7e54fcd0ebf4 (patch) | |
| tree | 9ff60356ad77b07c3704826c14eb316a9eb9ac27 /src/cpu/kernels/CpuCastKernel.h | |
| parent | 314d3e2c691734ff942c9a1aca1120b826a267d9 (diff) | |
| download | ComputeLibrary-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 'src/cpu/kernels/CpuCastKernel.h')
| -rw-r--r-- | src/cpu/kernels/CpuCastKernel.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/cpu/kernels/CpuCastKernel.h b/src/cpu/kernels/CpuCastKernel.h index de4ace2140..76237368d8 100644 --- a/src/cpu/kernels/CpuCastKernel.h +++ b/src/cpu/kernels/CpuCastKernel.h @@ -57,9 +57,10 @@ public: * - BFLOAT16 -> F32 * - F16 -> QASYMM8_SIGNED, QASYMM8, F32, S32, U8 * - S32 -> QASYMM8_SIGNED, QASYMM8, F16, F32, U8 + * - S64 -> F32 * - F32 -> QASYMM8_SIGNED, QASYMM8, BFLOAT16, F16, S32, U8 * - * @param[in] src The src tensor to convert. Data types supported: QASYMM8_SIGNED/QASYMM8/U8/U16/S16/BFLOAT16/F16/F32. + * @param[in] src The src tensor to convert. Data types supported: QASYMM8_SIGNED/QASYMM8/U8/U16/S16/S32/S64/BFLOAT16/F16/F32. * @param[out] dst The dst tensor. Data types supported: QASYMM8_SIGNED/QASYMM8/U8/U16/S16/U32/S32/BFLOAT16/F16/F32. * @param[in] policy Conversion policy. * |