diff options
author | Jan Eilers <jan.eilers@arm.com> | 2020-08-12 14:59:06 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-08-17 16:10:54 +0000 |
commit | 841aca155b35cc17ea9527599d2c364695e28166 (patch) | |
tree | 8861d6f275f5955495201f0e1b0677bb44604a17 /python/pyarmnn/test/test_descriptors.py | |
parent | 313c99f9a64c7fc51dc70757bffff3088c2e95cf (diff) | |
download | armnn-841aca155b35cc17ea9527599d2c364695e28166.tar.gz |
IVGCVSW-5200 Update pyarmnn
* Add HalfPixelCenters to Resize
* Update pyarmnn version to semantic versioning
* Add fill operator
* Add Bf16 optimization
* Add Gather operator
* Update TransposeConvolution2d descriptor
* Add Rank operator
* Add load dynamic tensor support of TfLiteParser
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: I7e76ed286ab87bd97a65ff62868ba7db7967376f
Diffstat (limited to 'python/pyarmnn/test/test_descriptors.py')
-rw-r--r-- | python/pyarmnn/test/test_descriptors.py | 21 |
1 files changed, 18 insertions, 3 deletions
diff --git a/python/pyarmnn/test/test_descriptors.py b/python/pyarmnn/test/test_descriptors.py index 6d49747d5a..b0574a14ba 100644 --- a/python/pyarmnn/test/test_descriptors.py +++ b/python/pyarmnn/test/test_descriptors.py @@ -143,6 +143,16 @@ def test_fakequantization_descriptor_default_values(): np.allclose(-6, desc.m_Min) +def test_fill_descriptor_default_values(): + desc = ann.FillDescriptor() + np.allclose(0, desc.m_Value) + + +def test_gather_descriptor_default_values(): + desc = ann.GatherDescriptor() + assert desc.m_Axis == 0 + + def test_fully_connected_descriptor_default_values(): desc = ann.FullyConnectedDescriptor() assert desc.m_BiasEnabled == False @@ -370,7 +380,7 @@ def test_space_to_batch_nd_descriptor_ctor(): def test_transpose_convolution2d_descriptor_default_values(): - desc = ann.DepthwiseConvolution2dDescriptor() + desc = ann.TransposeConvolution2dDescriptor() assert desc.m_PadLeft == 0 assert desc.m_PadTop == 0 assert desc.m_PadRight == 0 @@ -379,6 +389,7 @@ def test_transpose_convolution2d_descriptor_default_values(): assert desc.m_StrideY == 0 assert desc.m_BiasEnabled == False assert desc.m_DataLayout == ann.DataLayout_NCHW + assert desc.m_OutputShapeEnabled == False def test_view_descriptor_default_values(): @@ -480,7 +491,9 @@ generated_classes_names = list(map(lambda x: x[0], generated_classes)) 'StackDescriptor', 'StridedSliceDescriptor', 'TransposeConvolution2dDescriptor', - 'ElementwiseUnaryDescriptor']) + 'ElementwiseUnaryDescriptor', + 'FillDescriptor', + 'GatherDescriptor']) class TestDescriptorMassChecks: def test_desc_implemented(self, desc_name): @@ -522,7 +535,9 @@ generated_classes_names = list(map(lambda x: x[0], generated_classes)) 'StackDescriptor', 'StridedSliceDescriptor', 'TransposeConvolution2dDescriptor', - 'ElementwiseUnaryDescriptor']) + 'ElementwiseUnaryDescriptor', + 'FillDescriptor', + 'GatherDescriptor']) class TestDescriptorMassChecks: def test_desc_implemented(self, desc_name): |