aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/test/test_modeloption.py
diff options
context:
space:
mode:
authorFrancis Murtagh <francis.murtagh@arm.com>2022-06-21 13:16:23 +0000
committerFrancis Murtagh <francis.murtagh@arm.com>2022-06-22 08:36:28 +0000
commit626bd90378670eb5fd76f94526395430b752ad9e (patch)
treefa3c41a9ed7506f0bf5eb2b940f8661d2bf47adf /python/pyarmnn/test/test_modeloption.py
parentb9570c1a5b938aea20964e3b6a8164e9bf096b4c (diff)
downloadarmnn-626bd90378670eb5fd76f94526395430b752ad9e.tar.gz
Revert "Revert "IVGCVSW-6873 Import inputs but don't export outputs fails.""
This reverts commit a0f8b15d4ddb5075f380003ff31b271d389d3b66. Reason for revert: <Test ClDmaBufInternalTests review > Change-Id: Ibc4a77fa008643849da7330391942e4c87b941e2
Diffstat (limited to 'python/pyarmnn/test/test_modeloption.py')
-rw-r--r--python/pyarmnn/test/test_modeloption.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/python/pyarmnn/test/test_modeloption.py b/python/pyarmnn/test/test_modeloption.py
index c03d4a8cce..a47d2da358 100644
--- a/python/pyarmnn/test/test_modeloption.py
+++ b/python/pyarmnn/test/test_modeloption.py
@@ -71,7 +71,8 @@ def test_optimizer_options_with_model_opt():
False,
ShapeInferenceMethod_InferAndValidate,
True,
- [a])
+ [a],
+ True)
mo = oo.m_ModelOptions
@@ -112,7 +113,8 @@ def test_optimizer_options_fail():
False,
ShapeInferenceMethod_InferAndValidate,
True,
- a)
+ a,
+ True)
assert "Wrong number or type of arguments" in str(err.value)
@@ -122,7 +124,8 @@ def test_optimizer_options_fail():
True,
ShapeInferenceMethod_InferAndValidate,
True,
- [a])
+ [a],
+ True)
assert "BFloat16 and Float16 optimization cannot be enabled at the same time" in str(err.value)