ArmNN
 24.02
ChannelShuffleDescriptor Struct Reference

A ChannelShuffleDescriptor for the ChannelShuffle operator. More...

#include <Descriptors.hpp>

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

Public Member Functions

 ChannelShuffleDescriptor ()
 
 ChannelShuffleDescriptor (const uint32_t &numGroups, const uint32_t &axis)
 
bool operator== (const ChannelShuffleDescriptor &rhs) const
 
- Public Member Functions inherited from BaseDescriptor
virtual bool IsNull () const
 
virtual ~BaseDescriptor ()=default
 

Public Attributes

uint32_t m_NumGroups
 Number of groups for the channel shuffle operation. More...
 
uint32_t m_Axis
 Axis to apply channel shuffle operation on. More...
 

Detailed Description

A ChannelShuffleDescriptor for the ChannelShuffle operator.

Definition at line 1562 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ ChannelShuffleDescriptor() [1/2]

Definition at line 1564 of file Descriptors.hpp.

1565  : m_NumGroups(0), m_Axis(0)
1566  {}

◆ ChannelShuffleDescriptor() [2/2]

ChannelShuffleDescriptor ( const uint32_t &  numGroups,
const uint32_t &  axis 
)
inline

Definition at line 1568 of file Descriptors.hpp.

1569  : m_NumGroups(numGroups), m_Axis(axis)
1570  {}

Member Function Documentation

◆ operator==()

bool operator== ( const ChannelShuffleDescriptor rhs) const
inline

Definition at line 1572 of file Descriptors.hpp.

1573  {
1574  return m_NumGroups == rhs.m_NumGroups;
1575  }

References ChannelShuffleDescriptor::m_NumGroups.

Member Data Documentation

◆ m_Axis

◆ m_NumGroups

uint32_t m_NumGroups

Number of groups for the channel shuffle operation.

Definition at line 1578 of file Descriptors.hpp.

Referenced by ChannelShuffleDescriptor::operator==(), and StringifyLayerParameters< ChannelShuffleDescriptor >::Serialize().


The documentation for this struct was generated from the following file:
armnn::ChannelShuffleDescriptor::m_Axis
uint32_t m_Axis
Axis to apply channel shuffle operation on.
Definition: Descriptors.hpp:1580
armnn::ChannelShuffleDescriptor::m_NumGroups
uint32_t m_NumGroups
Number of groups for the channel shuffle operation.
Definition: Descriptors.hpp:1578