diff options
Diffstat (limited to 'verif/frameworks/tosa_verif_framework_generator.py')
| -rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 2a7d484..8ae0286 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -940,6 +940,7 @@ TF_OP_LIST = { tf.float32, ] }, + "num_variables": 2, }, "gru": { "operands": (1, 0), @@ -1445,6 +1446,12 @@ def run_unit_test( _, test_name = os.path.split(test_dir) + # For specifying the number of variable tensors if the graph has any + try: + num_varaibles = op["num_variables"] + except KeyError: + num_varaibles = 0 + # Write out test descriptor write_test_json( filename=os.path.join(test_dir, "test.json"), @@ -1461,6 +1468,7 @@ def run_unit_test( framework_exclusions=excluded_framework_list, quantized=is_quantized, test_name=test_name, + num_variables=num_varaibles, ) except Exception as e: msg = "Error running task: {}".format(e) |