ArmNN
 24.08
StackDescriptor Struct Reference

A StackDescriptor for the StackLayer. More...

#include <Descriptors.hpp>

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

Public Member Functions

 StackDescriptor ()
 
 StackDescriptor (uint32_t axis, uint32_t numInputs, const TensorShape &inputShape)
 
bool operator== (const StackDescriptor &rhs) const
 
- Public Member Functions inherited from BaseDescriptor
virtual bool IsNull () const
 
virtual ~BaseDescriptor ()=default
 

Public Attributes

uint32_t m_Axis
 0-based axis along which to stack the input tensors. More...
 
uint32_t m_NumInputs
 Number of input tensors. More...
 
TensorShape m_InputShape
 Required shape of all input tensors. More...
 

Detailed Description

A StackDescriptor for the StackLayer.

Definition at line 1251 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ StackDescriptor() [1/2]

StackDescriptor ( )
inline

Definition at line 1253 of file Descriptors.hpp.

1254  : m_Axis(0)
1255  , m_NumInputs(0)
1256  , m_InputShape()
1257  {}

◆ StackDescriptor() [2/2]

StackDescriptor ( uint32_t  axis,
uint32_t  numInputs,
const TensorShape inputShape 
)
inline

Definition at line 1259 of file Descriptors.hpp.

1260  : m_Axis(axis)
1261  , m_NumInputs(numInputs)
1262  , m_InputShape(inputShape)
1263  {}

Member Function Documentation

◆ operator==()

bool operator== ( const StackDescriptor rhs) const
inline

Definition at line 1265 of file Descriptors.hpp.

1266  {
1267  return m_Axis == rhs.m_Axis &&
1268  m_NumInputs == rhs.m_NumInputs &&
1269  m_InputShape == rhs.m_InputShape;
1270  }

References StackDescriptor::m_Axis, StackDescriptor::m_InputShape, and StackDescriptor::m_NumInputs.

Member Data Documentation

◆ m_Axis

◆ m_InputShape

◆ m_NumInputs


The documentation for this struct was generated from the following file:
armnn::StackDescriptor::m_NumInputs
uint32_t m_NumInputs
Number of input tensors.
Definition: Descriptors.hpp:1275
armnn::StackDescriptor::m_Axis
uint32_t m_Axis
0-based axis along which to stack the input tensors.
Definition: Descriptors.hpp:1273
armnn::StackDescriptor::m_InputShape
TensorShape m_InputShape
Required shape of all input tensors.
Definition: Descriptors.hpp:1277