ArmNN
 26.01
Loading...
Searching...
No Matches
TransposeDescriptor Struct Reference

A TransposeDescriptor for the TransposeLayer. More...

#include <Descriptors.hpp>

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

Public Member Functions

 TransposeDescriptor ()
 
 TransposeDescriptor (const PermutationVector &dimMappings)
 
bool operator== (const TransposeDescriptor &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.
 

Detailed Description

A TransposeDescriptor for the TransposeLayer.

Definition at line 1490 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ TransposeDescriptor() [1/2]

TransposeDescriptor ( )
inline

Definition at line 1492 of file Descriptors.hpp.

1493 : m_DimMappings{}
1494 {}
PermutationVector m_DimMappings
Indicates how to translate tensor elements from a given source into the target destination,...

◆ TransposeDescriptor() [2/2]

TransposeDescriptor ( const PermutationVector & dimMappings)
inline

Definition at line 1496 of file Descriptors.hpp.

1497 : m_DimMappings(dimMappings)
1498 {}

Member Function Documentation

◆ operator==()

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

Definition at line 1500 of file Descriptors.hpp.

1501 {
1502 return m_DimMappings.IsEqual(rhs.m_DimMappings);
1503 }
bool IsEqual(const PermutationVector &other) const
Definition Types.hpp:368

References PermutationVector::IsEqual(), and TransposeDescriptor::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, 4, 1} dim "0" of input goes into index 0 ([ 1, X, X, X]) dim "2" of input goes into index 1 ([ 1, 4, X, X ]) dim "3" of input goes into index 2 ([ 1, 4, 4, X ]) dim "1" of input goes into index 3 ([ 1, 4, 4, 1 ])

Definition at line 1514 of file Descriptors.hpp.

Referenced by armnn::ClTransposeWorkloadValidate(), ConvertTransposeToTosaOperator(), TransposeLayer::GetPermutation(), TransposeLayer::InferOutputShapes(), armnn::NeonTransposeWorkloadValidate(), TransposeDescriptor::operator==(), StringifyLayerParameters< TransposeDescriptor >::Serialize(), and TransposeQueueDescriptor::Validate().


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