24 #ifndef __ARM_COMPUTE_NEDIRECTCONVOLUTIONLAYERKERNEL_H__
25 #define __ARM_COMPUTE_NEDIRECTCONVOLUTIONLAYERKERNEL_H__
76 unsigned int _kernel_size;
77 unsigned int _num_weight_elems_read_per_row;
78 unsigned int _num_elems_read_per_iteration;
79 unsigned int _num_elems_written_per_iteration;
Container for 2D border size.
void configure(const ITensor *input, const ITensor *weights, ITensor *output, const PadStrideInfo &conv_info)
Set the input, weights, and output tensors.
Common interface for all kernels implemented in C++.
NEDirectConvolutionLayerKernel & operator=(const NEDirectConvolutionLayerKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
Interface for NEON tensor.
src info() -> set_format(Format::S16)
Padding and stride information class.
BorderSize border_size() const override
The size of the border for that kernel.
NEDirectConvolutionLayerKernel()
Default constructor.
~NEDirectConvolutionLayerKernel()=default
Default destructor.
NEON interface for Direct Convolution Layer kernel.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.