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. 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 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)
     {}

◆ 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 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;
     }