A StackDescriptor for the StackLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for StackDescriptor:

Collaboration diagram for StackDescriptor:

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.

         : m_Axis(0)
         , m_NumInputs(0)
         , m_InputShape()
     {}

◆ StackDescriptor() [2/2]

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

inline

Definition at line 1259 of file Descriptors.hpp.

         : m_Axis(axis)
         , m_NumInputs(numInputs)
         , m_InputShape(inputShape)
     {}

Member Function Documentation

◆ operator==()

bool operator== ( const StackDescriptor & rhs ) const

inline

Definition at line 1265 of file Descriptors.hpp.

     {
         return m_Axis       == rhs.m_Axis &&
                m_NumInputs  == rhs.m_NumInputs &&
                m_InputShape == rhs.m_InputShape;
     }