diff options
Diffstat (limited to 'python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i | 21 |
1 files changed, 20 insertions, 1 deletions
diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i index 8844cea81a..5f4afd399a 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i @@ -1000,7 +1000,6 @@ struct SoftmaxDescriptor bool operator ==(const SoftmaxDescriptor& rhs) const; }; - %feature("docstring", " A descriptor for the TransposeConvolution2d layer. See `INetwork.AddTransposeConvolution2dLayer()`. @@ -1056,6 +1055,26 @@ struct LogicalBinaryDescriptor bool operator ==(const LogicalBinaryDescriptor &rhs) const; }; +%feature("docstring", + " + A descriptor for the Transpose layer. See `INetwork.AddTransposeLayer()`. + + Contains: + m_DimMappings (PermutationVector): 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. {0U, 3U, 1U, 2U}. + + ") TransposeDescriptor; +struct TransposeDescriptor +{ + TransposeDescriptor(); + TransposeDescriptor(const PermutationVector& dimMappings); + + PermutationVector m_DimMappings; + + bool operator ==(const TransposeDescriptor &rhs) const; +}; + + using ConcatDescriptor = OriginsDescriptor; using LogSoftmaxDescriptor = SoftmaxDescriptor; using SplitterDescriptor = ViewsDescriptor; |