ArmNN
 25.11
Loading...
Searching...
No Matches
Pooling2dDescriptor Struct Reference

A Pooling2dDescriptor for the Pooling2dLayer. More...

#include <Descriptors.hpp>

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

Public Member Functions

 Pooling2dDescriptor ()
bool operator== (const Pooling2dDescriptor &rhs) const
Public Member Functions inherited from BaseDescriptor
virtual bool IsNull () const
virtual ~BaseDescriptor ()=default

Public Attributes

PoolingAlgorithm m_PoolType
 The pooling algorithm to use (Max. Average, L2).
uint32_t m_PadLeft
 Padding left value in the width dimension.
uint32_t m_PadRight
 Padding right value in the width dimension.
uint32_t m_PadTop
 Padding top value in the height dimension.
uint32_t m_PadBottom
 Padding bottom value in the height dimension.
uint32_t m_PoolWidth
 Pooling width value.
uint32_t m_PoolHeight
 Pooling height value.
uint32_t m_StrideX
 Stride value when proceeding through input for the width dimension.
uint32_t m_StrideY
 Stride value when proceeding through input for the height dimension.
OutputShapeRounding m_OutputShapeRounding
 The rounding method for the output shape. (Floor, Ceiling).
PaddingMethod m_PaddingMethod
 The padding method to be used. (Exclude, IgnoreValue).
DataLayout m_DataLayout
 The data layout to be used (NCHW, NHWC).

Detailed Description

A Pooling2dDescriptor for the Pooling2dLayer.

Definition at line 371 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ Pooling2dDescriptor()

Pooling2dDescriptor ( )
inline

Definition at line 373 of file Descriptors.hpp.

374 : m_PoolType(PoolingAlgorithm::Max)
375 , m_PadLeft(0)
376 , m_PadRight(0)
377 , m_PadTop(0)
378 , m_PadBottom(0)
379 , m_PoolWidth(0)
380 , m_PoolHeight(0)
381 , m_StrideX(1)
382 , m_StrideY(1)
383 , m_OutputShapeRounding(OutputShapeRounding::Floor)
384 , m_PaddingMethod(PaddingMethod::Exclude)
385 , m_DataLayout(DataLayout::NCHW)
386 {}

References armnn::Exclude, armnn::Floor, m_DataLayout, m_OutputShapeRounding, m_PadBottom, m_PaddingMethod, m_PadLeft, m_PadRight, m_PadTop, m_PoolHeight, m_PoolType, m_PoolWidth, m_StrideX, m_StrideY, armnn::Max, and armnn::NCHW.

Referenced by operator==().

Member Function Documentation

◆ operator==()

bool operator== ( const Pooling2dDescriptor & rhs) const
inline

Definition at line 388 of file Descriptors.hpp.

389 {
390 return m_PoolType == rhs.m_PoolType &&
391 m_PadLeft == rhs.m_PadLeft &&
392 m_PadRight == rhs.m_PadRight &&
393 m_PadTop == rhs.m_PadTop &&
394 m_PadBottom == rhs.m_PadBottom &&
395 m_PoolWidth == rhs.m_PoolWidth &&
396 m_PoolHeight == rhs.m_PoolHeight &&
397 m_StrideX == rhs.m_StrideX &&
398 m_StrideY == rhs.m_StrideY &&
399 m_OutputShapeRounding == rhs.m_OutputShapeRounding &&
400 m_PaddingMethod == rhs.m_PaddingMethod &&
401 m_DataLayout == rhs.m_DataLayout;
402 }

References m_DataLayout, m_OutputShapeRounding, m_PadBottom, m_PaddingMethod, m_PadLeft, m_PadRight, m_PadTop, m_PoolHeight, m_PoolType, m_PoolWidth, m_StrideX, m_StrideY, and Pooling2dDescriptor().

Member Data Documentation

◆ m_DataLayout

◆ m_OutputShapeRounding

OutputShapeRounding m_OutputShapeRounding

The rounding method for the output shape. (Floor, Ceiling).

Definition at line 423 of file Descriptors.hpp.

Referenced by IDeserializer::DeserializerImpl::GetPooling2dDescriptor(), operator==(), Pooling2dDescriptor(), and StringifyLayerParameters< Pooling2dDescriptor >::Serialize().

◆ m_PadBottom

◆ m_PaddingMethod

◆ m_PadLeft

◆ m_PadRight

◆ m_PadTop

◆ m_PoolHeight

◆ m_PoolType

◆ m_PoolWidth

◆ m_StrideX

◆ m_StrideY


The documentation for this struct was generated from the following file: