24 #ifndef __ARM_COMPUTE_CLIM2COLKERNEL_H__ 25 #define __ARM_COMPUTE_CLIM2COLKERNEL_H__ 105 void run_reduced(
const Window &window, cl::CommandQueue &queue);
111 void run_generic(
const Window &window, cl::CommandQueue &queue);
119 std::pair<unsigned int, unsigned int> _convolved_dims;
120 unsigned int _num_elems_processed_per_iteration;
121 Im2ColFunction _run_func;
Interface for the im2col reshape kernel.
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))
Static function to check if given info will lead to a valid configuration of CLIm2ColKernel.
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...
This file contains all available output stages for GEMMLowp on OpenCL.
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) ...
void configure(const ICLTensor *input, ICLTensor *output, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, const Size2D &dilation=Size2D(1U, 1U))
Set the input and output of the kernel.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.
CLIm2ColKernel()
Default constructor.