diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/pyarmnn/README.md | 1 | ||||
| -rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 42 | ||||
| -rw-r--r-- | python/pyarmnn/test/test_network.py | 3 |
3 files changed, 46 insertions, 0 deletions
diff --git a/python/pyarmnn/README.md b/python/pyarmnn/README.md index b510c361c7..a165f58609 100644 --- a/python/pyarmnn/README.md +++ b/python/pyarmnn/README.md @@ -1,5 +1,6 @@ # PyArmNN +## Note: Arm NN will be dropping support for PyArmNN in 24.08. PyArmNN is a python extension for [Arm NN SDK](https://developer.arm.com/ip-products/processors/machine-learning/arm-nn). PyArmNN provides interface similar to Arm NN C++ Api. Before you proceed with the project setup, you will need to checkout and build a corresponding Arm NN version. 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. diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index 91602b8fa8..933828329b 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', |