24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLCONV2D_H
25 #define ARM_COMPUTE_CLCONV2D_H
162 std::unique_ptr<IClOperator> _operator;
std::vector< MemoryInfo > MemoryRequirements
Convolution Layer Weights Information class.
SimpleTensor< float > src
static Status validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const Conv2dInfo &conv2d_info, const WeightsInfo &weights_info=WeightsInfo())
Static function to check if given info will lead to a valid configuration of ClConv2d.
ClConv2d & operator=(const ClConv2d &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
static ConvolutionMethod get_convolution_method(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *dst, const Conv2dInfo &conv2d_info, const WeightsInfo &weights_info, const GPUTarget gpu_target)
Static function to check if given info will return the convolution called by ClConv2d.
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
Descriptor used by the 2d Convolution function.
ConvolutionMethod
Available ConvolutionMethod.
Basic interface for functions which have a single async CL kernel.
~ClConv2d()
Default Destructor.
void prepare(ITensorPack &tensors) override
Prepare the function for executing.
ClConv2d()
Default constructor.
GPUTarget
Available GPU Targets.
void configure(const CLCompileContext &compile_context, ITensorInfo *src, ITensorInfo *weights, ITensorInfo *biases, ITensorInfo *dst, const Conv2dInfo &conv2d_info, const WeightsInfo &weights_info=WeightsInfo())
Set the src and dst tensors.
Basic function to compute the convolution layer.
Copyright (c) 2017-2024 Arm Limited.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Store the tensor's metadata.