diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-11-22 15:34:26 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2021-11-23 11:00:22 +0000 |
commit | 87dc8122b97f5e3b7f81b7434097b9be2db9c902 (patch) | |
tree | 409325d28960acdbb8de5446730e84a37bb11a90 /python/pyarmnn/src | |
parent | aeee9adbbee8c60fdfecc5bcc81b18e0748e4ee1 (diff) | |
download | armnn-87dc8122b97f5e3b7f81b7434097b9be2db9c902.tar.gz |
IVGCVSW-6588 AddCastLayer to PyArmNN
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: If9c6902396b9f3cec3b38b70ccfa12fcbb3fd5a7
Diffstat (limited to 'python/pyarmnn/src')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i index 0dc016fb96..3e3a36570e 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i @@ -427,6 +427,20 @@ public: armnn::IConnectableLayer* AddBatchToSpaceNdLayer(const armnn::BatchToSpaceNdDescriptor& batchToSpaceNdDescriptor, const char* name = nullptr); + + %feature("docstring", + " + Adds a Cast layer to the network. + + Args: + name (str): Optional name for the layer. + + Returns: + IConnectableLayer: Interface for configuring the layer. + ") AddCastLayer; + armnn::IConnectableLayer* AddCastLayer(const char* name = nullptr); + + %feature("docstring", " Adds a Comparison layer to the network. |