StackDescriptor Struct Reference

A StackDescriptor for the StackLayer. More...

#include <Descriptors.hpp>

Public Member Functions
	StackDescriptor ()
	StackDescriptor (uint32_t axis, uint32_t numInputs, const TensorShape &inputShape)
bool	operator== (const StackDescriptor &rhs) const

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

Constructor & Destructor Documentation

StackDescriptor() [1/2]

StackDescriptor ( )

inline

Definition at line 964 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 970 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 976 of file Descriptors.hpp.

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

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

Member Data Documentation

m_Axis

uint32_t m_Axis

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

Definition at line 984 of file Descriptors.hpp.

Referenced by ClStackWorkload::ClStackWorkload(), RefStackWorkload::Execute(), StackLayer::InferOutputShapes(), NeonStackWorkload::NeonStackWorkload(), StackDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), TfLiteParser::OutputShapeOfReshape(), StringifyLayerParameters< StackDescriptor >::Serialize(), armnn::Stack(), StackInferOutputShapeFromInputsMatchTest(), StackInferOutputShapeFromInputsNoMatchTest(), StackValidateTensorShapesFromInputsMatchTest(), StackValidateTensorShapesFromInputsNoMatchTest(), and SerializerVisitor::VisitStackLayer().

m_InputShape

TensorShape m_InputShape

Required shape of all input tensors.

Definition at line 988 of file Descriptors.hpp.

Referenced by ClStackWorkload::ClStackWorkload(), StackLayer::InferOutputShapes(), NeonStackWorkload::NeonStackWorkload(), StackDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), TfLiteParser::OutputShapeOfReshape(), StringifyLayerParameters< StackDescriptor >::Serialize(), StackInferOutputShapeFromInputsMatchTest(), StackInferOutputShapeFromInputsNoMatchTest(), StackValidateTensorShapesFromInputsMatchTest(), StackValidateTensorShapesFromInputsNoMatchTest(), StackLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitStackLayer().

m_NumInputs

uint32_t m_NumInputs

Number of input tensors.

Definition at line 986 of file Descriptors.hpp.

Referenced by RefStackWorkload::Execute(), StackLayer::InferOutputShapes(), StackDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), TfLiteParser::OutputShapeOfReshape(), StringifyLayerParameters< StackDescriptor >::Serialize(), StackInferOutputShapeFromInputsMatchTest(), StackInferOutputShapeFromInputsNoMatchTest(), StackValidateTensorShapesFromInputsMatchTest(), StackValidateTensorShapesFromInputsNoMatchTest(), StackLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitStackLayer().

---

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

• include/armnn/Descriptors.hpp