diff options
author | Nikhil Raj <nikhil.raj@arm.com> | 2024-05-14 14:16:40 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2024-05-14 16:19:31 +0100 |
commit | d7f483fcb1e5b07a7a59a3fad7be07c78a3abc37 (patch) | |
tree | 26d18b7b4fa39c1457f17804357cbf4f21f91abf | |
parent | 1358e04de341c4e50c8afcf4227990cce88b7526 (diff) | |
download | armnn-upstream-main.tar.gz |
IVGCVSW-8261 Ensure PyArmNN has been updated with new features added in ArmNNupstream-main
* Add layers missing in PyArmNN
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: Ia821105f9d235dcf7f9472e54849d268e07fb9a9
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 42 | ||||
-rw-r--r-- | python/pyarmnn/test/test_network.py | 3 |
2 files changed, 45 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 0b7f55d1c..bfaa55919 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. diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index 91602b8fa..933828329 100644 --- 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', |