commit 87dc8122b97f5e3b7f81b7434097b9be2db9c902
author Teresa Charlin <teresa.charlinreyes@arm.com> Mon Nov 22 15:34:26 2021 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Tue Nov 23 11:00:22 2021 +0000
tree 409325d28960acdbb8de5446730e84a37bb11a90
parent aeee9adbbee8c60fdfecc5bcc81b18e0748e4ee1

IVGCVSW-6588 AddCastLayer to PyArmNN

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: If9c6902396b9f3cec3b38b70ccfa12fcbb3fd5a7

python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i

diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
index 0dc016f..3e3a365 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 @@
     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.

python/pyarmnn/test/test_network.py

diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py
index f07f424..d734f61 100644
--- a/python/pyarmnn/test/test_network.py
+++ b/python/pyarmnn/test/test_network.py
@@ -192,6 +192,7 @@
     'AddArgMinMaxLayer',
     'AddBatchNormalizationLayer',
     'AddBatchToSpaceNdLayer',
+    'AddCastLayer',
     'AddComparisonLayer',
     'AddConcatLayer',
     'AddConstantLayer',