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

…in ArmNN * Add layers missing in PyArmNN Signed-off-by: Nikhil Raj <[email protected]> Change-Id: Ia821105f9d235dcf7f9472e54849d268e07fb9a9
ARM-software · May 15, 2024 · 1f3eb73 · 1f3eb73
1 parent 1999dac
commit 1f3eb73
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 0 deletions.
diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
@@ -596,6 +596,20 @@ public:
         const char* name = nullptr);
 
 
+    %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.
@@ -1147,6 +1161,20 @@ public:
     armnn::IConnectableLayer* AddFullyConnectedLayer(const armnn::FullyConnectedDescriptor& fullyConnectedDescriptor,
                                                      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.
@@ -1161,6 +1189,20 @@ public:
     armnn::IConnectableLayer* AddLogicalBinaryLayer(const armnn::LogicalBinaryDescriptor& logicalBinaryDescriptor,
                                                     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.

diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py
@@ -193,6 +193,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir):
     'AddBatchMatMulLayer',
     'AddBatchNormalizationLayer',
     'AddBatchToSpaceNdLayer',
+    'AddBroadcastToLayer',
     'AddCastLayer',
     'AddChannelShuffleLayer',
     'AddComparisonLayer',
@@ -210,6 +211,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir):
     'AddFloorLayer',
     'AddFillLayer',
     'AddFullyConnectedLayer',
+    'AddFusedLayer',
     'AddGatherLayer',
     'AddGatherNdLayer',
     'AddInputLayer',
@@ -237,6 +239,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir):
     'AddReshapeLayer',
     'AddResizeLayer',
     'AddReverseV2Layer',
+    'AddScatterNdLayer',
     'AddShapeLayer',
     'AddSliceLayer',
     'AddSoftmaxLayer',