24 #ifndef ARM_COMPUTE_NEDIRECTCONVOLUTIONLAYER_H 25 #define ARM_COMPUTE_NEDIRECTCONVOLUTIONLAYER_H 38 class NEDirectConvolutionLayerOutputStageKernel;
39 class NEDirectConvolutionLayerKernel;
40 class NEFillBorderKernel;
112 std::unique_ptr<NEDirectConvolutionLayerOutputStageKernel> _output_stage_kernel;
113 std::unique_ptr<NEDirectConvolutionLayerKernel> _conv_kernel;
114 std::unique_ptr<NEFillBorderKernel> _input_border_handler;
118 bool _is_activationlayer_enabled;
119 unsigned int _dim_split;
120 bool _is_padding_required;
Base class for all functions.
~NEDirectConvolutionLayer()
Default destructor.
void run() override
Run the kernels contained in the function.
Store the tensor's metadata.
Activation Layer Information class.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
NEDirectConvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Constructor.
Basic implementation of the tensor interface.
Padding and stride information class.
Basic function to run cpu::kernels::CpuActivationKernel.
void configure(ITensor *input, const ITensor *weights, const ITensor *bias, ITensor *output, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Set the input, weights, biases and output tensors.
Function to run the direct convolution.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *bias, 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 NEDirectConvolutionLayer...
NEDirectConvolutionLayer & operator=(const NEDirectConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...