From d5b1512b1d2cea3b87e52a0ecc123db2a7a7cad3 Mon Sep 17 00:00:00 2001 From: Jerry Ge Date: Tue, 26 Mar 2024 20:51:48 +0000 Subject: Add variable tensor fields for test descriptors - Add variable_name and variable_file to the desc.json file for writing variable tensors to numpy - Add the key of num_variables in the unit test declaration to specify the number of variable tensors in the graph Signed-off-by: Jerry Ge Change-Id: I1109f66ffed52e49dbb14f4a8aca64baa2bea622 --- verif/frameworks/tosa_verif_framework_generator.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'verif/frameworks/tosa_verif_framework_generator.py') 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) -- cgit v1.2.1