24 #ifndef __ARM_COMPUTE_NECONVOLUTIONLAYER_H__ 25 #define __ARM_COMPUTE_NECONVOLUTIONLAYER_H__ 79 Tensor _input_im2col_reshaped;
80 Tensor _input_interleaved_reshaped;
82 Tensor _weights_transposed;
Base class for all functions.
NEON kernel to perform reshaping on the weights used by convolution layer.
void run() override
Run the kernels contained in the function.
Interface for the im2col reshape kernel.
Interface for NEON tensor.
NEON kernel to perform col2im reshaping.
NEON kernel to interleave the elements of a matrix.
Basic function to simulate a convolution layer.
NEON kernel which transposes the elements of a matrix in chunks of 1x4 if the input data type is F32 ...
Basic implementation of the tensor interface.
Padding and stride information class.
void configure(const ITensor *input, const ITensor *weights, const ITensor *biases, ITensor *output, const PadStrideInfo &conv_info)
Set the input and output tensors.
NEConvolutionLayer()
Constructor.
NEON kernel to multiply two input matrices "A" and "B".