diff options
author | Ryan OShea <ryan.oshea3@arm.com> | 2023-07-28 17:35:18 +0100 |
---|---|---|
committer | ryan.oshea3 <ryan.oshea3@arm.com> | 2023-08-02 13:33:28 +0000 |
commit | 083802d04b7a4499c4daba860c57e4f152f9c060 (patch) | |
tree | d1f51625f1d5046e6127235dceecdab6705bf0ec /python/pyarmnn/test | |
parent | 7f2f59da45102cbb40df8d0a6b1159b787b96fde (diff) | |
download | armnn-083802d04b7a4499c4daba860c57e4f152f9c060.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: Iac70fb7f857978d676d3b67a4cc20fb5f3270676
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..91602b8fa8 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' ]) |