24.02
|
A TransposeDescriptor for the TransposeLayer. More...
#include <Descriptors.hpp>
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. More... | |
A TransposeDescriptor for the TransposeLayer.
Definition at line 1490 of file Descriptors.hpp.
|
inline |
Definition at line 1492 of file Descriptors.hpp.
|
inline |
Definition at line 1496 of file Descriptors.hpp.
|
inline |
Definition at line 1500 of file Descriptors.hpp.
References PermutationVector::IsEqual(), and TransposeDescriptor::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 ClTransposeWorkload::ClTransposeWorkload(), ConvertTransposeToTosaOperator(), TransposeLayer::GetPermutation(), TransposeLayer::InferOutputShapes(), NeonTransposeWorkload::NeonTransposeWorkload(), TransposeDescriptor::operator==(), StringifyLayerParameters< TransposeDescriptor >::Serialize(), and TransposeQueueDescriptor::Validate().