From 8a2d7cecea194ac8eafa91721fb5b09ae01e5971 Mon Sep 17 00:00:00 2001 From: Gunes Bayir Date: Wed, 28 Dec 2022 10:28:20 +0000 Subject: Extend Transposed Conv. for tiles with N0>1 Partially Resolves: COMPMID-5724 Signed-off-by: Gunes Bayir Change-Id: I0aeddddcdd87c8c79f6dae9a76ffdc2ba0c08e17 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8883 Benchmark: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Gian Marco Iodice Comments-Addressed: Arm Jenkins --- src/core/CL/cl_kernels/nhwc/transposed_convolution.cl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/core/CL') diff --git a/src/core/CL/cl_kernels/nhwc/transposed_convolution.cl b/src/core/CL/cl_kernels/nhwc/transposed_convolution.cl index 1ca282caea..fe6182fc95 100644 --- a/src/core/CL/cl_kernels/nhwc/transposed_convolution.cl +++ b/src/core/CL/cl_kernels/nhwc/transposed_convolution.cl @@ -52,8 +52,8 @@ * @note If bias exists, the compile time argument -DHAS_BIAS should be passed * @note Only the following configurations of M0, N0 and K0 are currently supported: * - M0 = 1 - * - N0 = 1 - * - K0 = 2, 3, 4, 8, 16 + * - N0 = 1, 2, 3, 4, 8, 16 + * - K0 = 1, 2, 3, 4, 8, 16 * * @note In case of QASYMM8/QASYMM8_SIGNED, the following extra information must be passed at compile time: * - -DIS_QUANTIZED -- cgit v1.2.1