IVGCVSW-6865 Add GATHERNd to PyArmNN
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I7c80299f539b64682a6a68cd1709089f71600cd9
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: