commit d7f483fcb1e5b07a7a59a3fad7be07c78a3abc37
author Nikhil Raj <nikhil.raj@arm.com> Tue May 14 14:16:40 2024 +0100
committer Nikhil Raj <nikhil.raj@arm.com> Tue May 14 16:19:31 2024 +0100
tree 26d18b7b4fa39c1457f17804357cbf4f21f91abf
parent 1358e04de341c4e50c8afcf4227990cce88b7526

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

* Add layers missing in PyArmNN

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

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 0b7f55d..bfaa559 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 @@
 
 
     %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 @@
                                                      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 @@
                                                     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.
 

python/pyarmnn/test/test_network.py

diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py
index 91602b8..9338283 100644
--- a/python/pyarmnn/test/test_network.py
+++ b/python/pyarmnn/test/test_network.py
@@ -193,6 +193,7 @@
     'AddBatchMatMulLayer',
     'AddBatchNormalizationLayer',
     'AddBatchToSpaceNdLayer',
+    'AddBroadcastToLayer',
     'AddCastLayer',
     'AddChannelShuffleLayer',
     'AddComparisonLayer',
@@ -210,6 +211,7 @@
     'AddFloorLayer',
     'AddFillLayer',
     'AddFullyConnectedLayer',
+    'AddFusedLayer',
     'AddGatherLayer',
     'AddGatherNdLayer',
     'AddInputLayer',
@@ -237,6 +239,7 @@
     'AddReshapeLayer',
     'AddResizeLayer',
     'AddReverseV2Layer',
+    'AddScatterNdLayer',
     'AddShapeLayer',
     'AddSliceLayer',
     'AddSoftmaxLayer',