diff options
| author | Ryan OShea <ryan.oshea3@arm.com> | 2021-11-18 16:52:41 +0000 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2021-11-26 10:27:39 +0000 |
| commit | 09a052251b3c4c14c093e79d83c36504746890be (patch) | |
| tree | db70f9e26250cf220db44265ec159878e53a1aa6 /python/pyarmnn/test | |
| parent | e0d1fc5ad4f47bfdff2fe2f3fbcebf68669257dd (diff) | |
| download | armnn-09a052251b3c4c14c093e79d83c36504746890be.tar.gz | |
IVGCVSW-6589 AddReduceLayer to PyArmNN
* AddReduceLayer to PyArmNN armnn_network.i
* AddReduceDescriptor to PyArmNN armnn_descriptors.i
* Add layer to test_network.py
* Add descriptor unit tests to test_descriptors.py
Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: I2a8706052baf8f9c0038a5ea327d73b8f3ea1961
Diffstat (limited to 'python/pyarmnn/test')
| -rw-r--r-- | python/pyarmnn/test/test_descriptors.py | 7 | ||||
| -rw-r--r-- | python/pyarmnn/test/test_network.py | 1 |
2 files changed, 8 insertions, 0 deletions
diff --git a/python/pyarmnn/test/test_descriptors.py b/python/pyarmnn/test/test_descriptors.py index 80c5359eb2..262b8fcf2a 100644 --- a/python/pyarmnn/test/test_descriptors.py +++ b/python/pyarmnn/test/test_descriptors.py @@ -288,6 +288,11 @@ def test_reshape_descriptor_default_values(): # check the empty Targetshape assert desc.m_TargetShape.GetNumDimensions() == 0 +def test_reduce_descriptor_default_values(): + desc = ann.ReduceDescriptor() + assert desc.m_KeepDims == False + assert desc.m_vAxis == [] + assert desc.m_ReduceOperation == ann.ReduceOperation_Sum def test_slice_descriptor_default_values(): desc = ann.SliceDescriptor() @@ -517,6 +522,7 @@ generated_classes_names = list(map(lambda x: x[0], generated_classes)) 'InstanceNormalizationDescriptor', 'BatchToSpaceNdDescriptor', 'FakeQuantizationDescriptor', + 'ReduceDescriptor', 'ResizeDescriptor', 'ReshapeDescriptor', 'SpaceToBatchNdDescriptor', @@ -565,6 +571,7 @@ generated_classes_names = list(map(lambda x: x[0], generated_classes)) 'InstanceNormalizationDescriptor', 'BatchToSpaceNdDescriptor', 'FakeQuantizationDescriptor', + 'ReduceDescriptor', 'ResizeDescriptor', 'ReshapeDescriptor', 'SpaceToBatchNdDescriptor', diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index e33470e090..e56621a23b 100644 --- a/python/pyarmnn/test/test_network.py +++ b/python/pyarmnn/test/test_network.py @@ -229,6 +229,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddQuantizeLayer', 'AddQuantizedLstmLayer', 'AddRankLayer', + 'AddReduceLayer', 'AddReshapeLayer', 'AddResizeLayer', 'AddShapeLayer', |