A ScatterNdDescriptor for the ScatterNdLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for ScatterNdDescriptor:

Collaboration diagram for ScatterNdDescriptor:

Public Member Functions
	ScatterNdDescriptor ()
	ScatterNdDescriptor (ScatterNdFunction function, bool inputEnabled)
	ScatterNdDescriptor (ScatterNdFunction function, bool inputEnabled, int32_t axis)
bool	operator== (const ScatterNdDescriptor &rhs) const
Public Member Functions inherited from BaseDescriptor
virtual bool	IsNull () const
virtual	~BaseDescriptor ()=default

Public Attributes
ScatterNdFunction	m_Function
	Specify if the function is update, add, sub, max or min.
bool	m_InputEnabled
	Flag to show if input tensor is accepted.
int32_t	m_Axis
	Extra attribute for ScatterElement, will be set to 0 by default, we do not support axis != 0.
bool	m_AxisEnabled
	Flag for ScatterElement, will be set to false by default, we do not support m_AxisEnable = true for now.

Detailed Description

A ScatterNdDescriptor for the ScatterNdLayer.

Definition at line 1679 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ ScatterNdDescriptor() [1/3]

ScatterNdDescriptor ( )

inline

Definition at line 1682 of file Descriptors.hpp.

            : m_Function(ScatterNdFunction::Update)
            , m_InputEnabled(true)
            , m_Axis(0)
            , m_AxisEnabled(false)
    {}

References m_Axis, m_AxisEnabled, m_Function, m_InputEnabled, and armnn::Update.

Referenced by operator==().

◆ ScatterNdDescriptor() [2/3]

ScatterNdDescriptor	(	ScatterNdFunction	function,
		bool	inputEnabled )

inline

Definition at line 1690 of file Descriptors.hpp.

            : m_Function(function)
            , m_InputEnabled(inputEnabled)
            , m_Axis(0)
            , m_AxisEnabled(false)
 
    {}

References m_Axis, m_AxisEnabled, m_Function, and m_InputEnabled.

◆ ScatterNdDescriptor() [3/3]

ScatterNdDescriptor	(	ScatterNdFunction	function,
		bool	inputEnabled,
		int32_t	axis )

inline

Definition at line 1700 of file Descriptors.hpp.

            : m_Function(function)
            , m_InputEnabled(inputEnabled)
            , m_Axis(axis)
            , m_AxisEnabled(true)
 
    {}

References m_Axis, m_AxisEnabled, m_Function, and m_InputEnabled.

Member Function Documentation

◆ operator==()

bool operator== ( const ScatterNdDescriptor & rhs ) const

inline

Definition at line 1710 of file Descriptors.hpp.

    {
        return ((m_Function == rhs.m_Function) &&
                (m_InputEnabled == rhs.m_InputEnabled)  &&
                (m_AxisEnabled == rhs.m_AxisEnabled) &&
                (m_Axis  == rhs.m_Axis));
    }