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.
uint32_t	m_NumInputs
	Number of input tensors.
TensorShape	m_InputShape
	Required shape of all input tensors.

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

References m_Axis, m_InputShape, and m_NumInputs.

Referenced by operator==().

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

References m_Axis, m_InputShape, and m_NumInputs.

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

References m_Axis, m_InputShape, m_NumInputs, and StackDescriptor().