ArmNN
 22.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:
BaseDescriptor

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 788 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ BatchNormalizationDescriptor()

BatchNormalizationDescriptor ( )
inline

Definition at line 790 of file Descriptors.hpp.

791  : m_Eps(0.0001f)
792  , m_DataLayout(DataLayout::NCHW)
793  {}
armnn::BatchNormalizationDescriptor::m_Eps
float m_Eps
Value to add to the variance. Used to avoid dividing by zero.
Definition: Descriptors.hpp:801
armnn::BatchNormalizationDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:803

Member Function Documentation

◆ operator==()

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

Definition at line 795 of file Descriptors.hpp.

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

796  {
797  return m_Eps == rhs.m_Eps && m_DataLayout == rhs.m_DataLayout;
798  }
armnn::BatchNormalizationDescriptor::m_Eps
float m_Eps
Value to add to the variance. Used to avoid dividing by zero.
Definition: Descriptors.hpp:801
armnn::BatchNormalizationDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:803

Member Data Documentation

◆ m_DataLayout

DataLayout m_DataLayout

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

Definition at line 803 of file Descriptors.hpp.

Referenced by armnn::BatchNormImpl(), TestBatchNormalizationLayerVisitor::CheckDescriptor(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), BatchNormalizationDescriptor::operator==(), StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize(), TEST_SUITE(), and armnn::TEST_SUITE().

◆ m_Eps

float m_Eps

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

Definition at line 801 of file Descriptors.hpp.

Referenced by armnn::BatchNormImpl(), TestBatchNormalizationLayerVisitor::CheckDescriptor(), CompareBatchNormTest(), OnnxParserImpl::CreateNetworkFromString(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), BatchNormalizationDescriptor::operator==(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize(), TEST_SUITE(), and armnn::TEST_SUITE().

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