diff options
Diffstat (limited to 'python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 17 |
1 files changed, 16 insertions, 1 deletions
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", |