24 #ifndef __ARM_COMPUTE_NEBATCHNORMALIZATIONLAYER_H__ 25 #define __ARM_COMPUTE_NEBATCHNORMALIZATIONLAYER_H__
Base class for all functions.
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.
This file contains all available output stages for GEMMLowp on OpenCL.
Basic function to run NENormalizationLayerKernel and simulate a batch normalization layer...
Interface for the batch normalization layer kernel.
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()
Default constructor.