commit 26ee542b19fc69ce975281cdf428813a073e393e
author Teresa Charlin <teresa.charlinreyes@arm.com> Tue May 03 21:39:57 2022 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Wed May 04 12:39:39 2022 +0000
tree 4697557f6dd8140986f419edcdf89a0449981925
parent d5c0ed24ce91ee0da1dcb5858da16f0f8a3d3172

IVGCVSW-6865 Add GATHERNd to PyArmNN


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

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 1957288..74ae8c1 100644
--- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
+++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
@@ -631,6 +631,18 @@
 
     %feature("docstring",
         "
+        Add GatherNd layer to the network.
+
+        Args:
+            name (str): Optional name for the layer.
+
+        Returns:
+            IConnectableLayer: Interface for configuring the layer.
+        ") AddGatherNdLayer;
+    armnn::IConnectableLayer* AddGatherNdLayer(const char* name = nullptr);
+
+    %feature("docstring",
+        "
         Adds an Instance Normalization 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 8cb8122..5522bf6 100644
--- a/python/pyarmnn/test/test_network.py
+++ b/python/pyarmnn/test/test_network.py
@@ -209,6 +209,7 @@
     'AddFillLayer',
     'AddFullyConnectedLayer',
     'AddGatherLayer',
+    'AddGatherNdLayer',
     'AddInputLayer',
     'AddInstanceNormalizationLayer',
     'AddLogicalBinaryLayer',