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/test_descriptors.py | |
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/test_descriptors.py')
-rw-r--r-- | python/pyarmnn/test/test_descriptors.py | 7 |
1 files changed, 7 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', |