24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_NEGEMMCONV2D_H
25 #define ARM_COMPUTE_NEGEMMCONV2D_H
53 NEGEMMConv2d(
const std::shared_ptr<IMemoryManager> &memory_manager =
nullptr);
118 std::unique_ptr<Impl> _impl;
Base class for all functions.
void configure(ITensor *input, const ITensor *weights, const ITensor *biases, ITensor *output, const Conv2dInfo &info)
Set the input and output tensors.
Interface for CPU tensor.
void prepare() override
Prepare the function for executing.
Descriptor used by the 2d Convolution function.
void run() override
Run the kernels contained in the function.
Basic function to compute the convolution layer.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const Conv2dInfo &info)
Static function to check if given info will lead to a valid configuration of NEGEMMConv2d.
NEGEMMConv2d & operator=(const NEGEMMConv2d &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
NEGEMMConv2d(const std::shared_ptr< IMemoryManager > &memory_manager=nullptr)
Constructor.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
~NEGEMMConv2d()
Destructor.