diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-21 10:03:49 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-21 10:05:11 +0100 |
commit | ea063dfe1533162581ee968ae44189edd3f78abc (patch) | |
tree | 4a2fff825792ddcee60716bedd33fb6ae82e3931 /python/pyarmnn/test/test_network.py | |
parent | 0fe7376f6982842e32591f3996d5ee650d22156f (diff) | |
download | armnn-ea063dfe1533162581ee968ae44189edd3f78abc.tar.gz |
IVGCVSW-5200 Add import enabled optimizer options to PyArmNN
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ic7c288fd829d7f1f1ae2910c47fbccdd635def8f
Diffstat (limited to 'python/pyarmnn/test/test_network.py')
-rw-r--r-- | python/pyarmnn/test/test_network.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index 679e640374..c24b113cdb 100644 --- a/python/pyarmnn/test/test_network.py +++ b/python/pyarmnn/test/test_network.py @@ -11,18 +11,28 @@ def test_optimizer_options_default_values(): assert opt.m_ReduceFp32ToFp16 == False assert opt.m_Debug == False assert opt.m_ReduceFp32ToBf16 == False + assert opt.m_ImportEnabled == 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 + assert opt.m_ImportEnabled == 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 + assert opt.m_ImportEnabled == False + +def test_optimizer_options_set_values3(): + opt = ann.OptimizerOptions(False, False, True, True) + assert opt.m_ReduceFp32ToFp16 == False + assert opt.m_Debug == False + assert opt.m_ReduceFp32ToBf16 == True + assert opt.m_ImportEnabled == True @pytest.fixture(scope="function") def get_runtime(shared_data_folder, network_file): |