diff options
author | Ryan OShea <ryan.oshea3@arm.com> | 2023-07-28 17:38:19 +0100 |
---|---|---|
committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2023-07-31 10:16:04 +0000 |
commit | 9a5e5b7032788249055f40edd763420f6bdba8fe (patch) | |
tree | 281cbc53b0542d03503611a1989a8b176f8f0521 /python/pyarmnn/test | |
parent | 4f9a6d63cd263789377b97cc409e0c4c6d707830 (diff) | |
download | armnn-9a5e5b7032788249055f40edd763420f6bdba8fe.tar.gz |
IVGCVSW-7703 Ensure PyArmNN has been updated with new
features added in ArmNN
* Adds BatchMatMul layer and descriptor to pyarmnn
* Adds ReverseV2 layer to pyarmnn
* Adds ElementWiseBinary layer and descriptor to pyarmnn
* Adds Tile layer and descriptor to pyarmnn
* Adds network test for each layer
Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: I07116d9e7b1eb6b6f8a687d8ba7cfbd11c912d0d
Diffstat (limited to 'python/pyarmnn/test')
-rw-r--r-- | python/pyarmnn/test/test_network.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index 88be5a8e7f..2fe8938978 100644 --- a/python/pyarmnn/test/test_network.py +++ b/python/pyarmnn/test/test_network.py @@ -190,6 +190,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddActivationLayer', 'AddAdditionLayer', 'AddArgMinMaxLayer', + 'AddBatchMatMulLayer', 'AddBatchNormalizationLayer', 'AddBatchToSpaceNdLayer', 'AddCastLayer', @@ -204,6 +205,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddDequantizeLayer', 'AddDetectionPostProcessLayer', 'AddDivisionLayer', + 'AddElementWiseBinaryLayer', 'AddElementwiseUnaryLayer', 'AddFloorLayer', 'AddFillLayer', @@ -234,6 +236,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddReduceLayer', 'AddReshapeLayer', 'AddResizeLayer', + 'AddReverseV2Layer', 'AddShapeLayer', 'AddSliceLayer', 'AddSoftmaxLayer', @@ -245,6 +248,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddStridedSliceLayer', 'AddSubtractionLayer', 'AddSwitchLayer', + 'AddTileLayer', 'AddTransposeConvolution2dLayer', 'AddTransposeLayer' ]) |