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 /python/pyarmnn/test | |
| parent | 1358e04de341c4e50c8afcf4227990cce88b7526 (diff) | |
| download | armnn-main.tar.gz | |
* Add layers missing in PyArmNN
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: Ia821105f9d235dcf7f9472e54849d268e07fb9a9
Diffstat (limited to 'python/pyarmnn/test')
| -rw-r--r-- | python/pyarmnn/test/test_network.py | 3 |
1 files changed, 3 insertions, 0 deletions
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', |