21.05
|
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 1032 of file Descriptors.hpp.
|
inline |
Definition at line 1034 of file Descriptors.hpp.
|
inline |
Definition at line 1040 of file Descriptors.hpp.
|
inline |
Definition at line 1046 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 1054 of file Descriptors.hpp.
Referenced by ClStackWorkload::ClStackWorkload(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), StackLayer::InferOutputShapes(), NeonStackWorkload::NeonStackWorkload(), StackDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfReshape(), StringifyLayerParameters< StackDescriptor >::Serialize(), armnn::Stack(), StackInferOutputShapeFromInputsMatchTest(), StackInferOutputShapeFromInputsNoMatchTest(), StackValidateTensorShapesFromInputsMatchTest(), and StackValidateTensorShapesFromInputsNoMatchTest().
TensorShape m_InputShape |
Required shape of all input tensors.
Definition at line 1058 of file Descriptors.hpp.
Referenced by ClStackWorkload::ClStackWorkload(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), StackLayer::InferOutputShapes(), NeonStackWorkload::NeonStackWorkload(), StackDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfReshape(), StringifyLayerParameters< StackDescriptor >::Serialize(), StackInferOutputShapeFromInputsMatchTest(), StackInferOutputShapeFromInputsNoMatchTest(), StackValidateTensorShapesFromInputsMatchTest(), StackValidateTensorShapesFromInputsNoMatchTest(), and StackLayer::ValidateTensorShapesFromInputs().
uint32_t m_NumInputs |
Number of input tensors.
Definition at line 1056 of file Descriptors.hpp.
Referenced by armnnSerializer::GetFlatBufferArgMinMaxFunction(), StackLayer::InferOutputShapes(), StackDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfReshape(), StringifyLayerParameters< StackDescriptor >::Serialize(), armnn::Stack(), StackInferOutputShapeFromInputsMatchTest(), StackInferOutputShapeFromInputsNoMatchTest(), StackValidateTensorShapesFromInputsMatchTest(), StackValidateTensorShapesFromInputsNoMatchTest(), and StackLayer::ValidateTensorShapesFromInputs().