From c1c366d13277bbb7dde8e59ef8fa92e62c6553a4 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 30 Mar 2020 18:09:24 +0100 Subject: COMPMID-3292: Clean tests for Im2ColKernel. Change-Id: I005ac542a94aeed8df806699d3d237ac8973d1e6 Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3003 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Sang-Hoon Park --- arm_compute/core/CL/kernels/CLIm2ColKernel.h | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'arm_compute') diff --git a/arm_compute/core/CL/kernels/CLIm2ColKernel.h b/arm_compute/core/CL/kernels/CLIm2ColKernel.h index 1ccac5b7e9..61f2a3d489 100644 --- a/arm_compute/core/CL/kernels/CLIm2ColKernel.h +++ b/arm_compute/core/CL/kernels/CLIm2ColKernel.h @@ -75,8 +75,11 @@ public: * @param[in] kernel_dims The kernel dimensions (width and height). * @param[in] conv_info Contains padding and stride information described in @ref PadStrideInfo. * @param[in] has_bias In case biases are provided expands the matrix with 1. + * This is valid only for non-quantized inputs. * @param[in] dilation (Optional) Dilation, in elements, across x and y. Defaults to (1, 1). - * @param[in] num_groups (Optional) Number of groups when performing a grouped convolution. num_groups != 1 is only supported for NCHW data layout + * @param[in] num_groups (Optional) Number of groups when performing a grouped convolution. + * Number of groups other than 1 is only supported for NCHW data layout. + * Number of groups should be multiple to the number of channels. */ void configure(const ICLTensor *input, ICLTensor *output, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, const Size2D &dilation = Size2D(1U, 1U), unsigned int num_groups = 1); @@ -105,8 +108,11 @@ public: * @param[in] kernel_dims The kernel dimensions (width and height). * @param[in] conv_info Contains padding and stride information described in @ref PadStrideInfo. * @param[in] has_bias In case biases are provided expands the matrix with 1. + * This is valid only for non-quantized inputs. * @param[in] dilation (Optional) Dilation, in elements, across x and y. Defaults to (1, 1). - * @param[in] num_groups (Optional) Number of groups when performing a grouped convolution. num_groups != 1 is only supported for NCHW data layout + * @param[in] num_groups (Optional) Number of groups when performing a grouped convolution. + * Number of groups other than 1 is only supported for NCHW data layout. + * Number of groups should be multiple to the number of channels. * * @return a status */ -- cgit v1.2.1