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. More...

bool	m_InputEnabled
	Flag to show if input tensor is accepted. More...

int32_t	m_Axis
	Extra attribute for ScatterElement, will be set to 0 by default, we do not support axis != 0. More...

bool	m_AxisEnabled
	Flag for ScatterElement, will be set to false by default, we do not support m_AxisEnable = true for now. More...

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 armnn::Update.

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

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

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));
     }