IVGCVSW-8261 Ensure PyArmNN has been updated with new features added in ArmNNHEADmain

* Add layers missing in PyArmNN

Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: Ia821105f9d235dcf7f9472e54849d268e07fb9a9

1 files changed, 42 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 0b7f55d1cc..bfaa55919b 100644
--- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
+++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
@@ -597,6 +597,20 @@ public:
 
 
     %feature("docstring",
+             "
+    Adds a Broadcast_To layer to the network.
+
+        Args:
+            broadcastToDescriptor (BroadcastToDescriptor): Descriptor for the explicit broadcat operation.
+            name (str): Optional name for the layer.
+
+            Returns:
+                IConnectableLayer: Interface for configuring the layer.
+    ") AddBroadcastToLayer;
+    armnn::IConnectableLayer* AddBroadcastToLayer(const armnn::BroadcastToDescriptor& broadcastToDescriptor,
+                                                  const char* name = nullptr);
+
+    %feature("docstring",
         "
         Adds a Division layer to the network.
 
@@ -1148,6 +1162,20 @@ public:
                                                      const char* name = nullptr);
 
     %feature("docstring",
+             "
+    Adds a Fused layer to the network. This is a precompiled layer for fused operator that merges Add + Mul + Add.
+
+        Args:
+            fusedDescriptor (FusedDescriptor): Description of the fused layer.
+            name (str): Optional name for the layer.
+
+            Returns:
+    IConnectableLayer: Interface for configuring the layer.
+    ") AddFusedLayer;
+    armnn::IConnectableLayer* AddFusedLayer(const armnn::FusedDescriptor& fusedDescriptor,
+                                            const char* name = nullptr);
+
+    %feature("docstring",
         "
         Adds a LogicalBinary layer to the network.
 
@@ -1162,6 +1190,20 @@ public:
                                                     const char* name = nullptr);
 
     %feature("docstring",
+             "
+    Adds a ScatterND layer to the network.
+
+            Args:
+    scatterndDescriptor (ScatterNdDescriptor): Description of the tile layer.
+            name (str): Optional name for the layer.
+
+            Returns:
+    IConnectableLayer: Interface for configuring the layer.
+    ") AddScatterNdLayer;
+    armnn::IConnectableLayer* AddScatterNdLayer(const armnn::ScatterNdDescriptor& scatterndDescriptor,
+                                                const char* name = nullptr);
+
+    %feature("docstring",
         "
         Adds a Tile layer to the network.