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_network.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_network.py')
-rw-r--r-- | python/pyarmnn/test/test_network.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index fc2591c1d5..679e640374 100644 --- a/python/pyarmnn/test/test_network.py +++ b/python/pyarmnn/test/test_network.py @@ -6,6 +6,23 @@ import stat import pytest import pyarmnn as ann +def test_optimizer_options_default_values(): + opt = ann.OptimizerOptions() + assert opt.m_ReduceFp32ToFp16 == False + assert opt.m_Debug == False + assert opt.m_ReduceFp32ToBf16 == False + +def test_optimizer_options_set_values1(): + opt = ann.OptimizerOptions(True, True) + assert opt.m_ReduceFp32ToFp16 == True + assert opt.m_Debug == True + assert opt.m_ReduceFp32ToBf16 == False + +def test_optimizer_options_set_values2(): + opt = ann.OptimizerOptions(False, False, True) + assert opt.m_ReduceFp32ToFp16 == False + assert opt.m_Debug == False + assert opt.m_ReduceFp32ToBf16 == True @pytest.fixture(scope="function") def get_runtime(shared_data_folder, network_file): @@ -166,6 +183,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddDivisionLayer', 'AddElementwiseUnaryLayer', 'AddFloorLayer', + 'AddFillLayer', 'AddFullyConnectedLayer', 'AddGatherLayer', 'AddInputLayer', @@ -186,6 +204,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddPreluLayer', 'AddQuantizeLayer', 'AddQuantizedLstmLayer', + 'AddRankLayer', 'AddReshapeLayer', 'AddResizeLayer', 'AddSliceLayer', |