ArmNN
 24.02
PermuteDescriptor Struct Reference

A PermuteDescriptor for the PermuteLayer. More...

#include <Descriptors.hpp>

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

Public Member Functions

 PermuteDescriptor ()
 
 PermuteDescriptor (const PermutationVector &dimMappings)
 
bool operator== (const PermuteDescriptor &rhs) const
 
- Public Member Functions inherited from BaseDescriptor
virtual bool IsNull () const
 
virtual ~BaseDescriptor ()=default
 

Public Attributes

PermutationVector m_DimMappings
 Indicates how to translate tensor elements from a given source into the target destination, when source and target potentially have different memory layouts e.g. More...
 

Detailed Description

A PermuteDescriptor for the PermuteLayer.

Definition at line 149 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ PermuteDescriptor() [1/2]

PermuteDescriptor ( )
inline

Definition at line 151 of file Descriptors.hpp.

152  : m_DimMappings{}
153  {}

◆ PermuteDescriptor() [2/2]

PermuteDescriptor ( const PermutationVector dimMappings)
inline

Definition at line 155 of file Descriptors.hpp.

156  : m_DimMappings(dimMappings)
157  {}

Member Function Documentation

◆ operator==()

bool operator== ( const PermuteDescriptor rhs) const
inline

Definition at line 159 of file Descriptors.hpp.

160  {
161  return m_DimMappings.IsEqual(rhs.m_DimMappings);
162  }

References PermutationVector::IsEqual(), and PermuteDescriptor::m_DimMappings.

Member Data Documentation

◆ m_DimMappings

PermutationVector m_DimMappings

Indicates how to translate tensor elements from a given source into the target destination, when source and target potentially have different memory layouts e.g.

Input Shape {1, 1, 4, 4} Permutation Vector {0, 2, 3, 1} Output Shape {1, 4, 1, 4} dim "0" goes into index 0 ([ 1, X, X, X ]) dim "1" goes into index 2 ([ 1, X, 1, X ]) dim "2" goes into index 3 ([ 1, X, 1, 4 ]) dim "3" goes into index 1 ([ 1, 4, 1, 4 ])

Definition at line 173 of file Descriptors.hpp.

Referenced by PermuteLayer::GetPermutation(), PermuteLayer::InferOutputShapes(), PermuteDescriptor::operator==(), StringifyLayerParameters< PermuteDescriptor >::Serialize(), and PermuteQueueDescriptor::Validate().


The documentation for this struct was generated from the following file:
armnn::PermutationVector::IsEqual
bool IsEqual(const PermutationVector &other) const
Definition: Types.hpp:366
armnn::PermuteDescriptor::m_DimMappings
PermutationVector m_DimMappings
Indicates how to translate tensor elements from a given source into the target destination,...
Definition: Descriptors.hpp:173