24 #ifndef ARM_COMPUTE_CLCONV2D_H 25 #define ARM_COMPUTE_CLCONV2D_H 147 std::unique_ptr<IClOperator> _operator;
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
ClConv2d()
Default constructor.
Basic function to compute the convolution layer.
Store the tensor's metadata.
void prepare(ITensorPack &tensors) override
Prepare the function for executing.
ConvolutionMethod
Available ConvolutionMethod.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
std::vector< MemoryInfo > MemoryRequirements
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Convolution Layer Weights Information class.
~ClConv2d()
Default Destructor.
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.
Descriptor used by the Convolution function.
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.
GPUTarget
Available GPU Targets.
Basic interface for functions which have a single async CL kernel.
ClConv2d & operator=(const ClConv2d &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
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.