commit aeee9adbbee8c60fdfecc5bcc81b18e0748e4ee1
author Ryan OShea <ryan.oshea3@arm.com> Thu Nov 18 17:43:55 2021 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Tue Nov 23 10:39:05 2021 +0000
tree 37e8a2a2ebd98ef90a256f3e2da6e75a05cbc0c2
parent 2b4182fa86c4efe43f3a631266b5185d8a725fa1

IVGCVSW-6587 AddShapeLayer to PyArmNN

 * Add layer to armnn.network.i
 * Add layer to test_network.py

Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: I7554a0ff62d81fcc99aa68204ae5f1cc3965c44b

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 f1cae56..0dc016f 100644
--- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
+++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
@@ -842,6 +842,18 @@
 
     %feature("docstring",
         "
+        Adds a Shape layer to the network.
+
+        Args:
+            name(str): Optional name for the layer.
+
+        Returns:
+            IConnectableLayer: Interface for configuring the layer
+        ") AddShapeLayer;
+    armnn::IConnectableLayer* AddShapeLayer(const char* name = nullptr);
+
+    %feature("docstring",
+        "
         Adds a Slice layer to the network.
 
         Args:

python/pyarmnn/test/test_network.py

diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py
index ff1c66e..f07f424 100644
--- a/python/pyarmnn/test/test_network.py
+++ b/python/pyarmnn/test/test_network.py
@@ -228,6 +228,7 @@
     'AddRankLayer',
     'AddReshapeLayer',
     'AddResizeLayer',
+    'AddShapeLayer',
     'AddSliceLayer',
     'AddSoftmaxLayer',
     'AddSpaceToBatchNdLayer',