24 #ifndef ARM_COMPUTE_CLIM2COLKERNEL_H 25 #define ARM_COMPUTE_CLIM2COLKERNEL_H 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)
Set the input and output of the kernel.
const Window & window() const
The maximum window the kernel can be executed on.
std::pair< unsigned int, unsigned int > _convolved_dims
Interface for the im2col reshape kernel.
Store the tensor's metadata.
Common interface for all the OpenCL kernels.
void run(const Window &window, cl::CommandQueue &queue) override
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue...
Copyright (c) 2017-2021 Arm Limited.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, const Size2D &dilation=Size2D(1U, 1U), unsigned int num_groups=1)
Static function to check if given info will lead to a valid configuration of CLIm2ColKernel.
const unsigned int num_groups
Padding and stride information class.
Interface for OpenCL tensor.
Class for specifying the size of an image or rectangle.
CLIm2ColKernel & operator=(const CLIm2ColKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
unsigned int _num_elems_processed_per_iteration
DataLayout
[DataLayout enum definition]
Describe a multidimensional execution window.
CLIm2ColKernel()
Default constructor.