24 #ifndef ARM_COMPUTE_CLCONVOLUTIONLAYER_H 25 #define ARM_COMPUTE_CLCONVOLUTIONLAYER_H 37 class CLCompileContext;
198 std::unique_ptr<Impl> _impl;
void run() override
Run the kernels contained in the function.
Base class for all functions.
CLConvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
void configure(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(), bool enable_fast_math=false, unsigned int num_groups=1)
Set the input and output tensors.
~CLConvolutionLayer()
Default Destructor.
void prepare() override
Prepare the function for executing.
Store the tensor's metadata.
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(), bool enable_fast_math=false, unsigned int num_groups=1)
Static function to check if given info will lead to a valid configuration of CLConvolutionLayer.
ConvolutionMethod
Available ConvolutionMethod.
Activation Layer Information class.
Copyright (c) 2017-2021 Arm Limited.
Convolution Layer Weights Information class.
const unsigned int num_groups
CLConvolutionLayer & operator=(const CLConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Padding and stride information class.
Basic function to compute the convolution layer.
Interface for OpenCL tensor.
GPUTarget
Available GPU Targets.
Class for specifying the size of an image or rectangle.
static ConvolutionMethod get_convolution_method(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info, const ActivationLayerInfo &act_info, const GPUTarget gpu_target, const Size2D &dilation=Size2D(1U, 1U), bool enable_fast_math=false)
Static function to check if given info will return the convolution called by CLConvolutionLayer.