24 #ifndef ARM_COMPUTE_NEDIRECTCONVOLUTIONLAYEROUTPUTSTAGEKERNEL_H 25 #define ARM_COMPUTE_NEDIRECTCONVOLUTIONLAYEROUTPUTSTAGEKERNEL_H 42 const char *
name()
const override 44 return "NEDirectConvolutionLayerOutputStageKernel";
90 int result_fixedpoint_multiplier,
int result_shift,
int result_offset_after_shift,
bool has_bias);
93 OutputStageKernel *_func;
97 int _result_fixedpoint_multiplier;
99 int _result_offset_after_shift;
const Window & window() const
The maximum window the kernel can be executed on.
Neon kernel to accumulate the biases, if provided, or downscale in case of quantized input...
Common interface for all kernels implemented in C++.
const char * name() const override
Name of the kernel.
Store the tensor's metadata.
Interface for Neon tensor.
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
NEDirectConvolutionLayerOutputStageKernel()
Default constructor.
Copyright (c) 2017-2021 Arm Limited.
void configure(ITensor *input, const ITensor *bias=nullptr, ITensor *output=nullptr, const DirectConvolutionLayerOutputStageKernelInfo &info=DirectConvolutionLayerOutputStageKernelInfo())
Set the accumulate buffer and the biases of the kernel.
NEDirectConvolutionLayerOutputStageKernel & operator=(const NEDirectConvolutionLayerOutputStageKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
static Status validate(const ITensorInfo *input, const ITensorInfo *bias=nullptr, const ITensorInfo *output=nullptr, const DirectConvolutionLayerOutputStageKernelInfo &info=DirectConvolutionLayerOutputStageKernelInfo())
Static function to check if given info will lead to a valid configuration of NEDirectConvolutionLayer...
~NEDirectConvolutionLayerOutputStageKernel()=default
Default destructor.
Information about executing thread and CPU.
Describe a multidimensional execution window.
Descriptor used by the direct convolution layer output stage kernels.