24 #ifndef ARM_COMPUTE_NEBATCHNORMALIZATIONLAYERKERNEL_H 25 #define ARM_COMPUTE_NEBATCHNORMALIZATIONLAYERKERNEL_H 39 const char *
name()
const override 41 return "NEBatchNormalizationLayerKernel";
97 void configure_non_fused();
99 void configure_fused();
109 template <
typename T,
bool fused_activation,
typename F>
110 void batch_normalization_nchw(
const Window &window);
119 template <
typename T,
bool fused_activation,
typename F>
120 void batch_normalization_nhwc(
const Window &window);
128 BatchNormFunctionPtr _func;
const char * name() const override
Name of the kernel.
const Window & window() const
The maximum window the kernel can be executed on.
Common interface for all kernels implemented in C++.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *mean, const ITensorInfo *var, const ITensorInfo *beta=nullptr, const ITensorInfo *gamma=nullptr, float epsilon=0.001f, ActivationLayerInfo act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of NEBatchNormalizationLaye...
Store the tensor's metadata.
Activation Layer Information class.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
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.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
Information about executing thread and CPU.
void configure(ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta=nullptr, const ITensor *gamma=nullptr, float epsilon=0.001f, ActivationLayerInfo act_info=ActivationLayerInfo())
Set the input and output tensors.
~NEBatchNormalizationLayerKernel()=default
Default destructor.
Describe a multidimensional execution window.