24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_ARM_COMPUTE_RUNTIME_CL_FUNCTIONS_CLCONVOLUTIONLAYER_H
25 #define ACL_ARM_COMPUTE_RUNTIME_CL_FUNCTIONS_CLCONVOLUTIONLAYER_H
38 class CLCompileContext;
131 bool enable_fast_math =
false,
162 bool enable_fast_math =
false,
193 bool enable_fast_math =
false,
222 bool enable_fast_math =
false);
229 std::unique_ptr<Impl> _impl;
232 #endif // ACL_ARM_COMPUTE_RUNTIME_CL_FUNCTIONS_CLCONVOLUTIONLAYER_H
Convolution Layer Weights Information class.
Base class for all functions.
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.
CLConvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
Basic function to compute the convolution layer.
Interface for OpenCL tensor.
Class for specifying the size of an image or rectangle.
void prepare() override
Prepare the function for executing.
ConvolutionMethod
Available ConvolutionMethod.
Activation Layer Information class.
void run() override
Run the kernels contained in the function.
~CLConvolutionLayer()
Default Destructor.
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.
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.
GPUTarget
Available GPU Targets.
const unsigned int num_groups
Copyright (c) 2017-2024 Arm Limited.
CLConvolutionLayer & operator=(const CLConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Store the tensor's metadata.