ArmNN
NotReleased
|
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... | |
A StackDescriptor for the StackLayer.
Definition at line 950 of file Descriptors.hpp.
|
inline |
Definition at line 952 of file Descriptors.hpp.
|
inline |
Definition at line 958 of file Descriptors.hpp.
|
inline |
Definition at line 964 of file Descriptors.hpp.
References StackDescriptor::m_Axis, StackDescriptor::m_InputShape, and StackDescriptor::m_NumInputs.
uint32_t m_Axis |
0-based axis along which to stack the input tensors.
Definition at line 972 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().
TensorShape m_InputShape |
Required shape of all input tensors.
Definition at line 976 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().
uint32_t m_NumInputs |
Number of input tensors.
Definition at line 974 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().