24.02.1
|
Go to the documentation of this file.
38 struct CLDirectConvolutionLayer::Impl
40 const ICLTensor *
src{
nullptr};
41 const ICLTensor *weights{
nullptr};
42 const ICLTensor *biases{
nullptr};
43 ICLTensor *
dst{
nullptr};
44 std::unique_ptr<opencl::ClDirectConv2d> op{
nullptr};
76 _impl->weights = weights;
77 _impl->biases = biases;
80 _impl->op = std::make_unique<opencl::ClDirectConv2d>();
81 _impl->op->configure(compile_context,
input->info(), weights->
info(),
102 _impl->op->run(
pack);
SimpleTensor< float > src
Interface for OpenCL tensor.
void add_tensor(int id, ITensor *tensor)
Add tensor to the pack.
static CLKernelLibrary & get()
Access the KernelLibrary singleton.
Activation Layer Information class.
#define ARM_COMPUTE_ERROR_ON_NULLPTR(...)
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
CLDirectConvolutionLayer & operator=(const CLDirectConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
void run() override
Run the kernels contained in the function.
Interface to enqueue OpenCL kernels and get/set the OpenCL CommandQueue and ICLTuner.
CLDirectConvolutionLayer()
Constructor.
~CLDirectConvolutionLayer()
Destructor.
static Status validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration.
Basic function to execute direct convolution function:
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
void configure(ICLTensor *input, const ICLTensor *weights, const ICLTensor *biases, ICLTensor *output, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Set the input and output tensors.
#define ARM_COMPUTE_LOG_PARAMS(...)
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of CLDirectConvolutionLayer...