diff options
Diffstat (limited to 'verif')
-rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index a3405dc..efb2bde 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -802,6 +802,7 @@ TF_OP_LIST = { 1, ), ], + "output_name": "Identity", }, "batch_to_space": { "operands": (1, 0), @@ -826,6 +827,7 @@ TF_OP_LIST = { "dynamic_shape_dim": [ (0,), ], + "output_name": "Identity", }, "space_to_depth": { "operands": (1, 0), @@ -1610,6 +1612,11 @@ def generate_op_tests(args, op_name, shape_list, result_name, filter, unit_test_ except KeyError: pass + try: + result_name = op["output_name"] + except KeyError: + pass + # populate non quantized unit test arguments for dtype in nonquantized_dtypes: |