24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_ARM_COMPUTE_RUNTIME_CL_FUNCTIONS_CLGEMMCONVOLUTIONLAYER_H
25 #define ACL_ARM_COMPUTE_RUNTIME_CL_FUNCTIONS_CLGEMMCONVOLUTIONLAYER_H
39 class CLCompileContext;
173 std::unique_ptr<Impl> _impl;
176 #endif // ACL_ARM_COMPUTE_RUNTIME_CL_FUNCTIONS_CLGEMMCONVOLUTIONLAYER_H
Convolution Layer Weights Information class.
Base class for all functions.
void run() override
Run the kernels contained in the function.
Interface for OpenCL tensor.
Class for specifying the size of an image or rectangle.
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.
void prepare() override
Prepare the function for executing.
Activation Layer Information class.
~CLGEMMConvolutionLayer()
Default destructor.
CLGEMMConvolutionLayer & operator=(const CLGEMMConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
const unsigned int num_groups
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.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
Weights manager interface to handle weights transformations.
Basic function to compute the convolution layer.
CLGEMMConvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr, IWeightsManager *weights_manager=nullptr)
Constructor.