A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for SpaceToBatchNdDescriptor:

Collaboration diagram for SpaceToBatchNdDescriptor:

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 Member Functions inherited from BaseDescriptor
virtual bool	IsNull () const
virtual	~BaseDescriptor ()=default

Public Attributes
std::vector< unsigned int >	m_BlockShape
	Block shape value.
std::vector< std::pair< unsigned int, unsigned int > >	m_PadList
	Specifies the padding values for the input dimension: heightPad{top, bottom} widthPad{left, right}.
DataLayout	m_DataLayout
	The data layout to be used (NCHW, NHWC).

Detailed Description

A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.

Definition at line 1043 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ SpaceToBatchNdDescriptor() [1/2]

SpaceToBatchNdDescriptor ( )

inline

Definition at line 1045 of file Descriptors.hpp.

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

References m_BlockShape.

Referenced by operator==().

◆ SpaceToBatchNdDescriptor() [2/2]

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

inline

Definition at line 1051 of file Descriptors.hpp.

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

References m_BlockShape, m_DataLayout, m_PadList, and armnn::NCHW.

Member Function Documentation

◆ operator==()

bool operator== ( const SpaceToBatchNdDescriptor & rhs ) const

inline

Definition at line 1058 of file Descriptors.hpp.

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

References m_BlockShape, m_DataLayout, m_PadList, and SpaceToBatchNdDescriptor().