24 #ifndef __ARM_COMPUTE_NEBATCHNORMALIZATIONLAYERKERNEL_H__ 25 #define __ARM_COMPUTE_NEBATCHNORMALIZATIONLAYERKERNEL_H__ 38 const char *
name()
const override 40 return "NEBatchNormalizationLayerKernel";
98 void configure_non_fused();
100 void configure_fused();
107 template <
bool fused_activation>
108 void batch_normalization_qs8(
const Window &window);
115 template <
bool fused_activation>
116 void batch_normalization_qs16(
const Window &window);
123 template <
bool fused_activation>
124 void batch_normalization_fp16(
const Window &window);
132 template <
bool fused_activation,
typename F>
133 void batch_normalization_fp32(
const Window &window);
141 BatchNormFunctionPtr _func;
const char * name() const override
Name of the kernel.
void configure(ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo act_info=ActivationLayerInfo())
Set the input and output tensors.
Common interface for all kernels implemented in C++.
Store the tensor's metadata.
Activation Layer Information class.
Interface for NEON tensor.
src info() -> set_format(Format::S16)
This file contains all available output stages for GEMMLowp on OpenCL.
NEBatchNormalizationLayerKernel()
Default constructor.
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
NEBatchNormalizationLayerKernel & operator=(const NEBatchNormalizationLayerKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Interface for the batch normalization layer kernel.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *mean, const ITensorInfo *var, const ITensorInfo *beta, const ITensorInfo *gamma, float epsilon, ActivationLayerInfo act_info)
Static function to check if given info will lead to a valid configuration of NEBatchNormalizationLaye...
~NEBatchNormalizationLayerKernel()=default
Default destructor.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.