StackDescriptor Struct Reference

A StackDescriptor for the StackLayer. More...

#include <Descriptors.hpp>

Inheritance 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 1218 of file Descriptors.hpp.

Constructor & Destructor Documentation

StackDescriptor() [1/2]

StackDescriptor ( )

inline

Definition at line 1220 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 1226 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 1232 of file Descriptors.hpp.

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

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

Member Data Documentation

m_Axis

uint32_t m_Axis

0-based axis along which to stack the input tensors.

Definition at line 1240 of file Descriptors.hpp.

Referenced by ClStackWorkload::ClStackWorkload(), StackLayer::InferOutputShapes(), NeonStackWorkload::NeonStackWorkload(), StackDescriptor::operator==(), StringifyLayerParameters< StackDescriptor >::Serialize(), armnn::Stack(), and StackQueueDescriptor::Validate().

m_InputShape

TensorShape m_InputShape

Required shape of all input tensors.

Definition at line 1244 of file Descriptors.hpp.

Referenced by ClStackWorkload::ClStackWorkload(), StackLayer::InferOutputShapes(), NeonStackWorkload::NeonStackWorkload(), StackDescriptor::operator==(), StringifyLayerParameters< StackDescriptor >::Serialize(), StackQueueDescriptor::Validate(), and StackLayer::ValidateTensorShapesFromInputs().

m_NumInputs

uint32_t m_NumInputs

Number of input tensors.

Definition at line 1242 of file Descriptors.hpp.

Referenced by StackLayer::InferOutputShapes(), StackDescriptor::operator==(), StringifyLayerParameters< StackDescriptor >::Serialize(), armnn::Stack(), StackQueueDescriptor::Validate(), and StackLayer::ValidateTensorShapesFromInputs().

---

The documentation for this struct was generated from the following file:

• include/armnn/Descriptors.hpp