ArmNN
 24.08
ScatterNdDescriptor Struct Reference

A ScatterNdDescriptor for the ScatterNdLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for ScatterNdDescriptor:
[legend]
Collaboration diagram for ScatterNdDescriptor:
[legend]

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.

1684  , m_InputEnabled(true)
1685  , m_Axis(0)
1686  , m_AxisEnabled(false)
1687  {}

References armnn::Update.

◆ ScatterNdDescriptor() [2/3]

ScatterNdDescriptor ( ScatterNdFunction  function,
bool  inputEnabled 
)
inline

Definition at line 1690 of file Descriptors.hpp.

1692  : m_Function(function)
1693  , m_InputEnabled(inputEnabled)
1694  , m_Axis(0)
1695  , m_AxisEnabled(false)
1696 
1697  {}

◆ ScatterNdDescriptor() [3/3]

ScatterNdDescriptor ( ScatterNdFunction  function,
bool  inputEnabled,
int32_t  axis 
)
inline

Definition at line 1700 of file Descriptors.hpp.

1703  : m_Function(function)
1704  , m_InputEnabled(inputEnabled)
1705  , m_Axis(axis)
1706  , m_AxisEnabled(true)
1707 
1708  {}

Member Function Documentation

◆ operator==()

bool operator== ( const ScatterNdDescriptor rhs) const
inline

Definition at line 1710 of file Descriptors.hpp.

1711  {
1712  return ((m_Function == rhs.m_Function) &&
1713  (m_InputEnabled == rhs.m_InputEnabled) &&
1714  (m_AxisEnabled == rhs.m_AxisEnabled) &&
1715  (m_Axis == rhs.m_Axis));
1716  }

References ScatterNdDescriptor::m_Axis, ScatterNdDescriptor::m_AxisEnabled, ScatterNdDescriptor::m_Function, and ScatterNdDescriptor::m_InputEnabled.

Member Data Documentation

◆ m_Axis

int32_t m_Axis

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

Definition at line 1725 of file Descriptors.hpp.

Referenced by ScatterNdDescriptor::operator==().

◆ m_AxisEnabled

bool m_AxisEnabled

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

Definition at line 1728 of file Descriptors.hpp.

Referenced by ScatterNdDescriptor::operator==(), and armnn::ScatterNd().

◆ m_Function

ScatterNdFunction m_Function

Specify if the function is update, add, sub, max or min.

Definition at line 1719 of file Descriptors.hpp.

Referenced by ScatterNdDescriptor::operator==(), and armnn::ScatterNd().

◆ m_InputEnabled

bool m_InputEnabled

The documentation for this struct was generated from the following file:
armnn::ScatterNdDescriptor::m_InputEnabled
bool m_InputEnabled
Flag to show if input tensor is accepted.
Definition: Descriptors.hpp:1722
armnn::ScatterNdDescriptor::m_AxisEnabled
bool m_AxisEnabled
Flag for ScatterElement, will be set to false by default, we do not support m_AxisEnable = true for n...
Definition: Descriptors.hpp:1728
armnn::ScatterNdDescriptor::m_Axis
int32_t m_Axis
Extra attribute for ScatterElement, will be set to 0 by default, we do not support axis !...
Definition: Descriptors.hpp:1725
armnn::ScatterNdFunction::Update
@ Update
armnn::ScatterNdDescriptor::m_Function
ScatterNdFunction m_Function
Specify if the function is update, add, sub, max or min.
Definition: Descriptors.hpp:1719