diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-11-18 15:24:50 +0000 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-11-23 14:44:59 +0000 |
| commit | f7b5011298367c6d635b17ad029c627076072198 (patch) | |
| tree | c11f1f5e9a84128da71dcd1e76f855fb4746aabf /python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | |
| parent | 181473307ca4e8745dbb0c8474f2615d1752b03a (diff) | |
| download | armnn-f7b5011298367c6d635b17ad029c627076072198.tar.gz | |
IVGCVSW-6585 AddChannelShuffleLayer to PyArmNN
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I2bfc54ea9aae78c60a66d7a5c39a33ca8a238e62
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", |