24 #ifndef ARM_COMPUTE_CLGEMMCONVOLUTIONLAYER_H 25 #define ARM_COMPUTE_CLGEMMCONVOLUTIONLAYER_H 38 class CLCompileContext;
150 std::unique_ptr<Impl> _impl;
CLGEMMConvolutionLayer & operator=(const CLGEMMConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Base class for all functions.
Basic function to compute the convolution layer.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info=WeightsInfo(), const Size2D &dilation=Size2D(1U, 1U), const ActivationLayerInfo &act_info=ActivationLayerInfo(), unsigned int num_groups=1)
Static function to check if given info will lead to a valid configuration of CLGEMMConvolutionLayer.
Store the tensor's metadata.
Activation Layer Information class.
Copyright (c) 2017-2021 Arm Limited.
Convolution Layer Weights Information class.
CLGEMMConvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr, IWeightsManager *weights_manager=nullptr)
Constructor.
const unsigned int num_groups
Padding and stride information class.
void run() override
Run the kernels contained in the function.
Weights manager interface to handle weights transformations.
void configure(const ICLTensor *input, const ICLTensor *weights, const ICLTensor *biases, ICLTensor *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info=WeightsInfo(), const Size2D &dilation=Size2D(1U, 1U), const ActivationLayerInfo &act_info=ActivationLayerInfo(), unsigned int num_groups=1)
Set the input and output tensors.
Interface for OpenCL tensor.
Class for specifying the size of an image or rectangle.
void prepare() override
Prepare the function for executing.
~CLGEMMConvolutionLayer()
Default destructor.