A NormalizationDescriptor for the NormalizationLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for NormalizationDescriptor:

Collaboration diagram for NormalizationDescriptor:

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

Public Attributes
NormalizationAlgorithmChannel	m_NormChannelType
	Normalization channel algorithm to use (Across, Within).
NormalizationAlgorithmMethod	m_NormMethodType
	Normalization method algorithm to use (LocalBrightness, LocalContrast).
uint32_t	m_NormSize
	Depth radius value.
float	m_Alpha
	Alpha value for the normalization equation.
float	m_Beta
	Beta value for the normalization equation.
float	m_K
	Kappa value used for the across channel normalization equation.
DataLayout	m_DataLayout
	The data layout to be used (NCHW, NHWC).

Detailed Description

A NormalizationDescriptor for the NormalizationLayer.

Definition at line 769 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ NormalizationDescriptor()

NormalizationDescriptor ( )

inline

Definition at line 771 of file Descriptors.hpp.

        : m_NormChannelType(NormalizationAlgorithmChannel::Across)
        , m_NormMethodType(NormalizationAlgorithmMethod::LocalBrightness)
        , m_NormSize(0)
        , m_Alpha(0.f)
        , m_Beta(0.f)
        , m_K(0.f)
        , m_DataLayout(DataLayout::NCHW)
    {}

References armnn::Across, armnn::LocalBrightness, m_Alpha, m_Beta, m_DataLayout, m_K, m_NormChannelType, m_NormMethodType, m_NormSize, and armnn::NCHW.

Referenced by operator==().

Member Function Documentation

◆ operator==()

bool operator== ( const NormalizationDescriptor & rhs ) const

inline

Definition at line 781 of file Descriptors.hpp.

    {
        return m_NormChannelType == rhs.m_NormChannelType &&
               m_NormMethodType  == rhs.m_NormMethodType &&
               m_NormSize        == rhs.m_NormSize &&
               m_Alpha           == rhs.m_Alpha &&
               m_Beta            == rhs.m_Beta &&
               m_K               == rhs.m_K &&
               m_DataLayout      == rhs.m_DataLayout;
    }