24 #ifndef __ARM_COMPUTE_NEIM2COLKERNEL_H__ 25 #define __ARM_COMPUTE_NEIM2COLKERNEL_H__ 91 void run_reduced(
const Window &window);
96 void run_generic(
const Window &window);
103 Im2ColFunctionPtr _func;
106 std::pair<unsigned int, unsigned int> _convolved_dims;
108 unsigned int _kernel_size;
Common interface for all kernels implemented in C++.
Interface for the im2col reshape kernel.
NEIm2ColKernel & operator=(const NEIm2ColKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Interface for NEON tensor.
NEIm2ColKernel()
Default constructor.
Padding and stride information class.
void configure(const ITensor *input, ITensor *output, std::pair< unsigned int, unsigned int > convolved_dims, const PadStrideInfo &conv_info, bool has_bias)
Set the input and output of the kernel.
void run(const Window &window) override
Execute the kernel on the passed window.
~NEIm2ColKernel()=default
Default destructor.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.