A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for BatchToSpaceNdDescriptor:

Collaboration diagram for BatchToSpaceNdDescriptor:

Public Member Functions
	BatchToSpaceNdDescriptor ()

	BatchToSpaceNdDescriptor (std::vector< unsigned int > blockShape, std::vector< std::pair< unsigned int, unsigned int >> crops)

bool	operator== (const BatchToSpaceNdDescriptor &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 values. More...

std::vector< std::pair< unsigned int, unsigned int > >	m_Crops
	The values to crop from the input dimension. More...

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

Detailed Description

A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.

Definition at line 875 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ BatchToSpaceNdDescriptor() [1/2]

BatchToSpaceNdDescriptor ( )

inline

Definition at line 877 of file Descriptors.hpp.

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

◆ BatchToSpaceNdDescriptor() [2/2]

BatchToSpaceNdDescriptor	(	std::vector< unsigned int >	blockShape,
		std::vector< std::pair< unsigned int, unsigned int >>	crops
	)

inline

Definition at line 883 of file Descriptors.hpp.

         : m_BlockShape(blockShape)
         , m_Crops(crops)
         , m_DataLayout(DataLayout::NCHW)
     {}

References armnn::NCHW.

Member Function Documentation

◆ operator==()

bool operator== ( const BatchToSpaceNdDescriptor & rhs ) const

inline

Definition at line 890 of file Descriptors.hpp.

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