ArmNN
 25.11
Loading...
Searching...
No Matches
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.
DataLayout m_DataLayout
 The data layout to be used (NCHW, NHWC).

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 {}

References m_DataLayout, m_Eps, and armnn::NCHW.

Referenced by operator==().

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 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 BatchNormalizationDescriptor(), armnn::BatchNormImpl(), armnn::ClBatchNormalizationValidate(), armnn::NeonBatchNormalizationValidate(), 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 BatchNormalizationDescriptor(), armnn::BatchNormImpl(), armnn::ClBatchNormalizationValidate(), armnn::NeonBatchNormalizationValidate(), operator==(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), and StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize().

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