diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-05-03 21:39:57 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-05-04 12:39:39 +0000 |
commit | 26ee542b19fc69ce975281cdf428813a073e393e (patch) | |
tree | 4697557f6dd8140986f419edcdf89a0449981925 | |
parent | d5c0ed24ce91ee0da1dcb5858da16f0f8a3d3172 (diff) | |
download | armnn-26ee542b19fc69ce975281cdf428813a073e393e.tar.gz |
IVGCVSW-6865 Add GATHERNd to PyArmNN
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I7c80299f539b64682a6a68cd1709089f71600cd9
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 12 | ||||
-rw-r--r-- | python/pyarmnn/test/test_network.py | 1 |
2 files changed, 13 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 1957288003..74ae8c1cd2 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i @@ -631,6 +631,18 @@ public: %feature("docstring", " + Add GatherNd layer to the network. + + Args: + name (str): Optional name for the layer. + + Returns: + IConnectableLayer: Interface for configuring the layer. + ") AddGatherNdLayer; + armnn::IConnectableLayer* AddGatherNdLayer(const char* name = nullptr); + + %feature("docstring", + " Adds an Instance Normalization layer to the network. Args: diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index 8cb81221e2..5522bf656d 100644 --- a/python/pyarmnn/test/test_network.py +++ b/python/pyarmnn/test/test_network.py @@ -209,6 +209,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddFillLayer', 'AddFullyConnectedLayer', 'AddGatherLayer', + 'AddGatherNdLayer', 'AddInputLayer', 'AddInstanceNormalizationLayer', 'AddLogicalBinaryLayer', |