24 #ifndef ARM_COMPUTE_NEBATCHNORMALIZATIONLAYER_H 25 #define ARM_COMPUTE_NEBATCHNORMALIZATIONLAYER_H 36 class NEBatchNormalizationLayerKernel;
100 std::unique_ptr<NEBatchNormalizationLayerKernel> _norm_kernel;
Base class for all functions.
~NEBatchNormalizationLayer()
Default destructor.
Store the tensor's metadata.
Activation Layer Information class.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
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...
Basic function to run NENormalizationLayerKernel and simulate a batch normalization layer...
NEBatchNormalizationLayer & operator=(const NEBatchNormalizationLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
void run() override
Run the kernels contained in the function.
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.
NEBatchNormalizationLayer()
Constructor.