diff options
Diffstat (limited to 'verif/frameworks/write_test_json.py')
-rw-r--r-- | verif/frameworks/write_test_json.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/verif/frameworks/write_test_json.py b/verif/frameworks/write_test_json.py index 68dfc8f..d52a777 100644 --- a/verif/frameworks/write_test_json.py +++ b/verif/frameworks/write_test_json.py @@ -18,10 +18,14 @@ def write_test_json( ifm_shape=None, framework_exclusions=None, quantized=False, + test_name=None, ): test_desc = dict() + if test_name: + test_desc["name"] = test_name + if tf_model_filename: test_desc["tf_model_filename"] = tf_model_filename |