diff options
Diffstat (limited to 'python/pyarmnn/src')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/__init__.py | 4 | ||||
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i | 20 | ||||
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 17 |
3 files changed, 38 insertions, 3 deletions
diff --git a/python/pyarmnn/src/pyarmnn/__init__.py b/python/pyarmnn/src/pyarmnn/__init__.py index 1a71844cfc..b71fc3632b 100644 --- a/python/pyarmnn/src/pyarmnn/__init__.py +++ b/python/pyarmnn/src/pyarmnn/__init__.py @@ -79,8 +79,8 @@ from ._generated.pyarmnn import ActivationFunction_Abs, ActivationFunction_Bound ActivationFunction_Sqrt, ActivationFunction_Square, ActivationFunction_TanH, ActivationDescriptor from ._generated.pyarmnn import ArgMinMaxFunction_Max, ArgMinMaxFunction_Min, ArgMinMaxDescriptor from ._generated.pyarmnn import BatchNormalizationDescriptor, BatchToSpaceNdDescriptor -from ._generated.pyarmnn import ComparisonDescriptor, ComparisonOperation_Equal, ComparisonOperation_Greater, \ - ComparisonOperation_GreaterOrEqual, ComparisonOperation_Less, \ +from ._generated.pyarmnn import ChannelShuffleDescriptor, ComparisonDescriptor, ComparisonOperation_Equal, \ + ComparisonOperation_Greater, ComparisonOperation_GreaterOrEqual, ComparisonOperation_Less, \ ComparisonOperation_LessOrEqual, ComparisonOperation_NotEqual from ._generated.pyarmnn import UnaryOperation_Abs, UnaryOperation_Exp, UnaryOperation_Sqrt, UnaryOperation_Rsqrt, \ UnaryOperation_Neg, ElementwiseUnaryDescriptor diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i index e51c8674ab..d20796a88e 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i @@ -133,6 +133,26 @@ struct BatchToSpaceNdDescriptor %feature("docstring", " + A descriptor for the ChannelShuffle layer. See `INetwork.AddChannelShuffleLayer()`. + + Contains: + m_NumGroups (int): Underlying C++ type is uint32_t. Number of groups for the shuffle operation. Default: 0. + m_Axis (int): Underlying C++ type is uint32_t. 0-based axis along which shuffle is performed. Default: 0. + + ") ChannelShuffleDescriptor; +struct ChannelShuffleDescriptor +{ + ChannelShuffleDescriptor(); + ChannelShuffleDescriptor(int numGroups, int axis); + + int m_NumGroups; + int m_Axis; + + bool operator ==(const ChannelShuffleDescriptor &rhs) const; +}; + +%feature("docstring", + " A descriptor for the Comparison layer. See `INetwork.AddComparisonLayer()`. Contains: diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i index fe626dc2ea..789e428e46 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i @@ -425,7 +425,22 @@ public: IConnectableLayer: Interface for configuring the layer. ") AddBatchToSpaceNdLayer; armnn::IConnectableLayer* AddBatchToSpaceNdLayer(const armnn::BatchToSpaceNdDescriptor& batchToSpaceNdDescriptor, - const char* name = nullptr); + const char* name = nullptr); + + %feature("docstring", + " + Adds a ChannelShuffle layer to the network. + + Args: + channelShuffleDescriptor (ChannelShuffleDescriptor): Configuration parameters for the layer. + name (str): Optional name for the layer. + + Returns: + IConnectableLayer: Interface for configuring the layer. + ") AddChannelShuffleLayer; + armnn::IConnectableLayer* AddChannelShuffleLayer(const armnn::ChannelShuffleDescriptor& channelShuffleDescriptor, + const char* name = nullptr); + %feature("docstring", |