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

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