24 #ifndef ARM_COMPUTE_NEIM2COLKERNEL_H 25 #define ARM_COMPUTE_NEIM2COLKERNEL_H 60 const char *
name()
const override 62 return "NEIm2ColKernel";
118 template <
typename T,
bool has_pads,
bool is_nchw>
119 void run_im2col(
const Window &window);
127 Im2ColFunctionPtr _func;
130 std::pair<unsigned int, unsigned int> _convolved_dims;
132 unsigned int _kernel_width;
133 unsigned int _kernel_height;
const Window & window() const
The maximum window the kernel can be executed on.
const char * name() const override
Name of the kernel.
Common interface for all kernels implemented in C++.
Interface for the im2col reshape kernel.
Store the tensor's metadata.
NEIm2ColKernel & operator=(const NEIm2ColKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
const unsigned int num_groups
NEIm2ColKernel()
Default constructor.
Padding and stride information class.
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
Information about executing thread and CPU.
~NEIm2ColKernel()=default
Default destructor.
Class for specifying the size of an image or rectangle.
void configure(const ITensor *input, ITensor *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.
DataLayout
[DataLayout enum definition]
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 NEIm2ColKernel.
Describe a multidimensional execution window.