ArmNN
NotReleased
|
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer. More...
#include <Descriptors.hpp>
Public Member Functions | |
SpaceToBatchNdDescriptor () | |
SpaceToBatchNdDescriptor (const std::vector< unsigned int > &blockShape, const std::vector< std::pair< unsigned int, unsigned int >> &padList) | |
bool | operator== (const SpaceToBatchNdDescriptor &rhs) const |
Public Attributes | |
std::vector< unsigned int > | m_BlockShape |
Block shape value. More... | |
std::vector< std::pair< unsigned int, unsigned int > > | m_PadList |
Specifies the padding values for the input dimension: heightPad{top, bottom} widthPad{left, right}. More... | |
DataLayout | m_DataLayout |
The data layout to be used (NCHW, NHWC). More... | |
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.
Definition at line 778 of file Descriptors.hpp.
|
inline |
Definition at line 780 of file Descriptors.hpp.
References armnn::NCHW.
|
inline |
Definition at line 786 of file Descriptors.hpp.
|
inline |
Definition at line 793 of file Descriptors.hpp.
References SpaceToBatchNdDescriptor::m_BlockShape, SpaceToBatchNdDescriptor::m_DataLayout, and SpaceToBatchNdDescriptor::m_PadList.
std::vector<unsigned int> m_BlockShape |
Block shape value.
Definition at line 801 of file Descriptors.hpp.
Referenced by ClSpaceToBatchNdWorkload::ClSpaceToBatchNdWorkload(), TfLiteParser::CreateNetworkFromBinary(), SpaceToBatchNdLayer::CreateWorkload(), SpaceToBatchNdLayer::InferOutputShapes(), NeonSpaceToBatchNdWorkload::NeonSpaceToBatchNdWorkload(), SpaceToBatchNdDescriptor::operator==(), Deserializer::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToBatchNdDescriptor >::Serialize(), armnn::SpaceToBatchNd(), and SerializerVisitor::VisitSpaceToBatchNdLayer().
DataLayout m_DataLayout |
The data layout to be used (NCHW, NHWC).
Definition at line 806 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), TfLiteParser::CreateNetworkFromBinary(), SpaceToBatchNdLayer::InferOutputShapes(), SpaceToBatchNdDescriptor::operator==(), Deserializer::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToBatchNdDescriptor >::Serialize(), armnn::SpaceToBatchNd(), and SerializerVisitor::VisitSpaceToBatchNdLayer().
std::vector<std::pair<unsigned int, unsigned int> > m_PadList |
Specifies the padding values for the input dimension: heightPad{top, bottom} widthPad{left, right}.
Definition at line 804 of file Descriptors.hpp.
Referenced by TfLiteParser::CreateNetworkFromBinary(), SpaceToBatchNdLayer::CreateWorkload(), SpaceToBatchNdLayer::InferOutputShapes(), SpaceToBatchNdDescriptor::operator==(), Deserializer::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToBatchNdDescriptor >::Serialize(), armnn::SpaceToBatchNd(), and SerializerVisitor::VisitSpaceToBatchNdLayer().