ArmNN
 25.02
Public Member Functions | Public Attributes | List of all members
BatchNormalizationDescriptor Struct Reference

A BatchNormalizationDescriptor for the BatchNormalizationLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for BatchNormalizationDescriptor:
[legend]
Collaboration diagram for BatchNormalizationDescriptor:
[legend]

Public Member Functions

 BatchNormalizationDescriptor ()
 
bool operator== (const BatchNormalizationDescriptor &rhs) const
 
- Public Member Functions inherited from BaseDescriptor
virtual bool IsNull () const
 
virtual ~BaseDescriptor ()=default
 

Public Attributes

float m_Eps
 Value to add to the variance. Used to avoid dividing by zero. More...
 
DataLayout m_DataLayout
 The data layout to be used (NCHW, NHWC). More...
 

Detailed Description

A BatchNormalizationDescriptor for the BatchNormalizationLayer.

Definition at line 828 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ BatchNormalizationDescriptor()

BatchNormalizationDescriptor ( )
inline

Definition at line 830 of file Descriptors.hpp.

831  : m_Eps(0.0001f)
832  , m_DataLayout(DataLayout::NCHW)
833  {}
armnn::BatchNormalizationDescriptor::m_Eps
float m_Eps
Value to add to the variance. Used to avoid dividing by zero.
Definition: Descriptors.hpp:841
armnn::BatchNormalizationDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:843

References armnn::NCHW.

Member Function Documentation

◆ operator==()

bool operator== ( const BatchNormalizationDescriptor &  rhs) const
inline

Definition at line 835 of file Descriptors.hpp.

836  {
837  return m_Eps == rhs.m_Eps && m_DataLayout == rhs.m_DataLayout;
838  }

References BatchNormalizationDescriptor::m_DataLayout, and BatchNormalizationDescriptor::m_Eps.

Member Data Documentation

◆ m_DataLayout

DataLayout m_DataLayout

The data layout to be used (NCHW, NHWC).

Definition at line 843 of file Descriptors.hpp.

Referenced by armnn::BatchNormImpl(), BatchNormalizationDescriptor::operator==(), and StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize().

◆ m_Eps

float m_Eps

Value to add to the variance. Used to avoid dividing by zero.

Definition at line 841 of file Descriptors.hpp.

Referenced by armnn::BatchNormImpl(), BatchNormalizationDescriptor::operator==(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), and StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize().

The documentation for this struct was generated from the following file: