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 | |
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
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 14 | ||||
-rw-r--r-- | python/pyarmnn/test/test_network.py | 1 |
2 files changed, 15 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. diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index f07f424d52..d734f619f7 100644 --- a/python/pyarmnn/test/test_network.py +++ b/python/pyarmnn/test/test_network.py @@ -192,6 +192,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddArgMinMaxLayer', 'AddBatchNormalizationLayer', 'AddBatchToSpaceNdLayer', + 'AddCastLayer', 'AddComparisonLayer', 'AddConcatLayer', 'AddConstantLayer', |