ArmNN
 24.02
SpaceToBatchNdDescriptor Struct Reference

A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer. More...

#include <Descriptors.hpp>

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

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]

Definition at line 1045 of file Descriptors.hpp.

1046  : m_BlockShape({1, 1})
1047  , m_PadList({{0, 0}, {0, 0}})
1049  {}

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

1053  : m_BlockShape(blockShape)
1054  , m_PadList(padList)
1056  {}

References armnn::NCHW.

Member Function Documentation

◆ operator==()

bool operator== ( const SpaceToBatchNdDescriptor rhs) const
inline

Definition at line 1058 of file Descriptors.hpp.

1059  {
1060  return m_BlockShape == rhs.m_BlockShape &&
1061  m_PadList == rhs.m_PadList &&
1062  m_DataLayout == rhs.m_DataLayout;
1063  }

References SpaceToBatchNdDescriptor::m_BlockShape, SpaceToBatchNdDescriptor::m_DataLayout, and SpaceToBatchNdDescriptor::m_PadList.

Member Data Documentation

◆ m_BlockShape

◆ m_DataLayout

◆ m_PadList

std::vector<std::pair<unsigned int, unsigned int> > m_PadList

The documentation for this struct was generated from the following file:
armnn::SpaceToBatchNdDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:1071
armnn::SpaceToBatchNdDescriptor::m_BlockShape
std::vector< unsigned int > m_BlockShape
Block shape value.
Definition: Descriptors.hpp:1066
armnn::SpaceToBatchNdDescriptor::m_PadList
std::vector< std::pair< unsigned int, unsigned int > > m_PadList
Specifies the padding values for the input dimension: heightPad{top, bottom} widthPad{left,...
Definition: Descriptors.hpp:1069
armnn::DataLayout::NCHW
@ NCHW