24 #ifndef ARM_COMPUTE_NEGEMMCONV2D_H 25 #define ARM_COMPUTE_NEGEMMCONV2D_H 39 class NEGEMMAssemblyDispatch;
55 NEGEMMConv2d(
const std::shared_ptr<IMemoryManager> &memory_manager =
nullptr);
102 std::unique_ptr<NEGEMMAssemblyDispatch> _gemm_asm_func;
105 const ITensor *_original_weights;
108 bool _run_activation;
Base class for all functions.
~NEGEMMConv2d()
Destructor.
Store the tensor's metadata.
Basic function to run cpu::kernels::CpuPermuteKernel.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
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) ...
Basic implementation of the tensor interface.
Descriptor used by the Convolution function.
Basic function to run cpu::kernels::CpuActivationKernel.
void run() override
Run the kernels contained in the function.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
NEGEMMConv2d(const std::shared_ptr< IMemoryManager > &memory_manager=nullptr)
Constructor.
Basic function to compute the convolution layer.
void prepare() override
Prepare the function for executing.
void configure(ITensor *input, const ITensor *weights, const ITensor *biases, ITensor *output, const Conv2dInfo &info)
Set the input and output tensors.