SpaceToBatchNdDescriptor Struct Reference

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...

Detailed Description

A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.

Definition at line 835 of file Descriptors.hpp.

Constructor & Destructor Documentation

SpaceToBatchNdDescriptor() [1/2]

SpaceToBatchNdDescriptor ( )

inline

Definition at line 837 of file Descriptors.hpp.

References armnn::NCHW.

        : m_BlockShape({1, 1})
        , m_PadList({{0, 0}, {0, 0}})
        , m_DataLayout(DataLayout::NCHW)
    {}

SpaceToBatchNdDescriptor() [2/2]

SpaceToBatchNdDescriptor ( const std::vector< unsigned int > &  blockShape, const std::vector< std::pair< unsigned int, unsigned int >> &  padList  )

inline

Definition at line 843 of file Descriptors.hpp.

        : m_BlockShape(blockShape)
        , m_PadList(padList)
        , m_DataLayout(DataLayout::NCHW)
    {}

Member Function Documentation

operator==()

bool operator== ( const SpaceToBatchNdDescriptor &  rhs ) const

inline

Definition at line 850 of file Descriptors.hpp.

References SpaceToBatchNdDescriptor::m_BlockShape, SpaceToBatchNdDescriptor::m_DataLayout, and SpaceToBatchNdDescriptor::m_PadList.

    {
        return m_BlockShape == rhs.m_BlockShape &&
               m_PadList    == rhs.m_PadList &&
               m_DataLayout == rhs.m_DataLayout;
    }

Member Data Documentation

m_BlockShape

std::vector<unsigned int> m_BlockShape

Block shape value.

Definition at line 858 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().

m_DataLayout

DataLayout m_DataLayout

The data layout to be used (NCHW, NHWC).

Definition at line 863 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().

m_PadList

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

Referenced by TfLiteParser::CreateNetworkFromBinary(), SpaceToBatchNdLayer::CreateWorkload(), SpaceToBatchNdLayer::InferOutputShapes(), SpaceToBatchNdDescriptor::operator==(), Deserializer::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToBatchNdDescriptor >::Serialize(), armnn::SpaceToBatchNd(), and SerializerVisitor::VisitSpaceToBatchNdLayer().

---

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

• include/armnn/Descriptors.hpp