diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-03-30 18:09:24 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-04-28 14:50:46 +0000 |
| commit | c1c366d13277bbb7dde8e59ef8fa92e62c6553a4 (patch) | |
| tree | f5a2ee35451977f2e93f842e52f196ad3cc6e261 /arm_compute | |
| parent | 55c0c0c9559c08baed7f1c783404b83dff8c8e89 (diff) | |
| download | ComputeLibrary-c1c366d13277bbb7dde8e59ef8fa92e62c6553a4.tar.gz | |
COMPMID-3292: Clean tests for Im2ColKernel.
Change-Id: I005ac542a94aeed8df806699d3d237ac8973d1e6
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3003
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/core/CL/kernels/CLIm2ColKernel.h | 10 |
1 files changed, 8 insertions, 2 deletions
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 */ |