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

A BatchNormalizationDescriptor for the BatchNormalizationLayer. More...

#include <Descriptors.hpp>

Public Member Functions

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

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

Constructor & Destructor Documentation

◆ BatchNormalizationDescriptor()

BatchNormalizationDescriptor ( )
inline

Definition at line 620 of file Descriptors.hpp.

621  : m_Eps(0.0001f)
622  , m_DataLayout(DataLayout::NCHW)
623  {}
armnn::BatchNormalizationDescriptor::m_Eps
float m_Eps
Value to add to the variance. Used to avoid dividing by zero.
Definition: Descriptors.hpp:631
armnn::BatchNormalizationDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:633

Member Function Documentation

◆ operator==()

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

Definition at line 625 of file Descriptors.hpp.

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

626  {
627  return m_Eps == rhs.m_Eps && m_DataLayout == rhs.m_DataLayout;
628  }
armnn::BatchNormalizationDescriptor::m_Eps
float m_Eps
Value to add to the variance. Used to avoid dividing by zero.
Definition: Descriptors.hpp:631
armnn::BatchNormalizationDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:633

Member Data Documentation

◆ m_DataLayout

DataLayout m_DataLayout

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

Definition at line 633 of file Descriptors.hpp.

Referenced by armnn::BatchNormImpl(), armnn::BOOST_AUTO_TEST_CASE(), TestBatchNormalizationLayerVisitor::CheckDescriptor(), Deserializer::GetNetworkOutputBindingInfo(), BatchNormalizationDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize(), and SerializerVisitor::VisitBatchNormalizationLayer().

◆ m_Eps

float m_Eps

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

Definition at line 631 of file Descriptors.hpp.

Referenced by armnn::BatchNormImpl(), BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestBatchNormalizationLayerVisitor::CheckDescriptor(), CompareBatchNormTest(), OnnxParser::CreateNetworkFromString(), Deserializer::GetNetworkOutputBindingInfo(), BatchNormalizationDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), CaffeParserBase::ParseBatchNormLayer(), CaffeParserBase::ParseScaleLayer(), StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize(), and SerializerVisitor::VisitBatchNormalizationLayer().

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