diff options
author | alexander <alexander.efremov@arm.com> | 2021-10-18 19:17:24 +0100 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2021-10-20 13:05:51 +0000 |
commit | 73010788725f8f07efb6df20711ece712ee213ea (patch) | |
tree | c31261ee0d4bb69fdb76f4f5147b82a74763ff55 /python/pyarmnn/test/test_network.py | |
parent | e8d7ccb1a18b5c39ecf17fb063fcc50baacc74b3 (diff) | |
download | armnn-73010788725f8f07efb6df20711ece712ee213ea.tar.gz |
MLECO-2488: added model optimization options and updated OptimizerOptions constructor.
Signed-off-by: alexander <alexander.efremov@arm.com>
Change-Id: Ic2ad6a46c3830f2526ba8b20ca0db0780be4b9a2
Diffstat (limited to 'python/pyarmnn/test/test_network.py')
-rw-r--r-- | python/pyarmnn/test/test_network.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index c24b113cdb..9b2dbf7f3a 100644 --- a/python/pyarmnn/test/test_network.py +++ b/python/pyarmnn/test/test_network.py @@ -6,12 +6,15 @@ 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 assert opt.m_ImportEnabled == False + assert opt.m_shapeInferenceMethod == ann.ShapeInferenceMethod_ValidateOnly + def test_optimizer_options_set_values1(): opt = ann.OptimizerOptions(True, True) @@ -19,6 +22,8 @@ def test_optimizer_options_set_values1(): assert opt.m_Debug == True assert opt.m_ReduceFp32ToBf16 == False assert opt.m_ImportEnabled == False + assert opt.m_shapeInferenceMethod == ann.ShapeInferenceMethod_ValidateOnly + def test_optimizer_options_set_values2(): opt = ann.OptimizerOptions(False, False, True) @@ -26,13 +31,17 @@ def test_optimizer_options_set_values2(): assert opt.m_Debug == False assert opt.m_ReduceFp32ToBf16 == True assert opt.m_ImportEnabled == False + assert opt.m_shapeInferenceMethod == ann.ShapeInferenceMethod_ValidateOnly + def test_optimizer_options_set_values3(): - opt = ann.OptimizerOptions(False, False, True, True) + opt = ann.OptimizerOptions(False, False, True, ann.ShapeInferenceMethod_InferAndValidate, True) assert opt.m_ReduceFp32ToFp16 == False assert opt.m_Debug == False assert opt.m_ReduceFp32ToBf16 == True assert opt.m_ImportEnabled == True + assert opt.m_shapeInferenceMethod == ann.ShapeInferenceMethod_InferAndValidate + @pytest.fixture(scope="function") def get_runtime(shared_data_folder, network_file): |