24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLDECONVOLUTIONLAYER_H
25 #define ARM_COMPUTE_CLDECONVOLUTIONLAYER_H
126 std::shared_ptr<IMemoryManager> _memory_manager;
127 std::unique_ptr<IFunction> _function;
130 std::unique_ptr<Impl> _impl;
Basic function to compute the deconvolution layer.
Convolution Layer Weights Information class.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *bias, ITensorInfo *output, const PadStrideInfo &deconv_info, const WeightsInfo &weights_info=WeightsInfo())
Static function to check if given info will lead to a valid configuration of CLDeconvolutionLayer.
Base class for all functions.
DeconvolutionMethod
Available DeconvolutionMethod.
Interface for OpenCL tensor.
void run() override
Run the kernels contained in the function.
void prepare() override
Prepare the function for executing.
CLDeconvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
void configure(ICLTensor *input, ICLTensor *weights, const ICLTensor *bias, ICLTensor *output, const PadStrideInfo &deconv_info, const WeightsInfo &weights_info=WeightsInfo())
Set the input, weights, biases and output tensors.
static DeconvolutionMethod get_deconvolution_method(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *bias, ITensorInfo *output, const PadStrideInfo &deconv_info, const WeightsInfo &weights_info)