24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLDEPTHWISECONVOLUTION_H
25 #define ARM_COMPUTE_CLDEPTHWISECONVOLUTION_H
36 class CLCompileContext;
37 class CLDepthwiseConvolutionLayerNativeKernel;
97 unsigned int depth_multiplier = 1,
110 unsigned int depth_multiplier = 1,
125 unsigned int depth_multiplier = 1,
135 _memory_group =
MemoryGroup(std::move(memory_manager));
141 std::unique_ptr<CLDepthwiseConvolutionLayerNativeKernel> _dwc_native_kernel;
151 const ITensor *_original_weights;
~CLDepthwiseConvolutionLayer()
Default destructor.
Function to execute a depthwise convolution.
Base class for all functions.
Interface for OpenCL tensor.
Class for specifying the size of an image or rectangle.
Interface for CPU tensor.
Basic implementation of the OpenCL tensor interface.
Activation Layer Information class.
void prepare() override
Prepare the function for executing.
CLDepthwiseConvolutionLayer & operator=(const CLDepthwiseConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Basic function to execute an opencl::kernels::ClPermuteKernel.
void run() override
Run the kernels contained in the function.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const PadStrideInfo &conv_info, unsigned int depth_multiplier=1, ActivationLayerInfo act_info=ActivationLayerInfo(), const Size2D &dilation=Size2D(1U, 1U))
Static function to check if given info will lead to a valid configuration of CLDepthwiseConvolutionLa...
Copyright (c) 2017-2024 Arm Limited.
void set_memory_group(std::shared_ptr< IMemoryManager > memory_manager)
void configure(const CLCompileContext &compile_context, ICLTensor *input, const ICLTensor *weights, const ICLTensor *biases, ICLTensor *output, const PadStrideInfo &conv_info, unsigned int depth_multiplier=1, ActivationLayerInfo act_info=ActivationLayerInfo(), const Size2D &dilation=Size2D(1U, 1U))
Initialize the function's source, destination, weights and convolution information.
Store the tensor's metadata.
CLDepthwiseConvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.