24 #ifndef __ARM_COMPUTE_CLCONVOLUTIONLAYER_H__ 25 #define __ARM_COMPUTE_CLCONVOLUTIONLAYER_H__ 83 CLTensor _input_interleaved_reshaped;
OpenCL kernel which interleaves the elements of a matrix A in chunk of 4x4.
CLConvolutionLayer()
Default constructor.
void run() override
Run the kernels contained in the function.
Base class for all functions.
Interface for the weights reshape kernel used by convolution and fully connected layers.
OpenCL kernel which transposes the elements of a matrix in chunks of 1x4 if the input data type is F3...
Interface for the im2col reshape kernel.
OpenCL kernel to multiply two input matrices "A" and "B" or to multiply a vector "A" by a matrix "B"...
Interface for the col2im reshaping kernel.
Padding and stride information class.
void configure(const ICLTensor *input, const ICLTensor *weights, const ICLTensor *biases, ICLTensor *output, const PadStrideInfo &conv_info)
Set the input and output tensors.
Basic function to compute the convolution layer.
Interface for OpenCL tensor.
Basic implementation of the OpenCL tensor interface.