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.
std::vector< std::pair< unsigned int, unsigned int > >	m_Crops
	The values to crop from the input dimension.
DataLayout	m_DataLayout
	The data layout to be used (NCHW, NHWC).

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

References m_BlockShape.

Referenced by operator==().

◆ 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 m_BlockShape, m_Crops, m_DataLayout, and 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;
    }

References BatchToSpaceNdDescriptor(), m_BlockShape, m_Crops, and m_DataLayout.