diff options
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', |