ArmNN
 24.02
BatchToSpaceNdDescriptor Struct Reference

A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for BatchToSpaceNdDescriptor:
[legend]
Collaboration diagram for BatchToSpaceNdDescriptor:
[legend]

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]

Definition at line 877 of file Descriptors.hpp.

878  : m_BlockShape({1, 1})
879  , m_Crops({{0, 0}, {0, 0}})
881  {}

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

885  : m_BlockShape(blockShape)
886  , m_Crops(crops)
888  {}

References armnn::NCHW.

Member Function Documentation

◆ operator==()

bool operator== ( const BatchToSpaceNdDescriptor rhs) const
inline

Definition at line 890 of file Descriptors.hpp.

891  {
892  return m_BlockShape == rhs.m_BlockShape &&
893  m_Crops == rhs.m_Crops &&
894  m_DataLayout == rhs.m_DataLayout;
895  }

References BatchToSpaceNdDescriptor::m_BlockShape, BatchToSpaceNdDescriptor::m_Crops, and BatchToSpaceNdDescriptor::m_DataLayout.

Member Data Documentation

◆ m_BlockShape

◆ m_Crops

◆ m_DataLayout


The documentation for this struct was generated from the following file:
armnn::BatchToSpaceNdDescriptor::m_BlockShape
std::vector< unsigned int > m_BlockShape
Block shape values.
Definition: Descriptors.hpp:898
armnn::BatchToSpaceNdDescriptor::m_Crops
std::vector< std::pair< unsigned int, unsigned int > > m_Crops
The values to crop from the input dimension.
Definition: Descriptors.hpp:900
armnn::BatchToSpaceNdDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:902
armnn::DataLayout::NCHW
@ NCHW