diff options
| author | Jerry Ge <jerry.ge@arm.com> | 2024-05-30 21:21:16 +0000 |
|---|---|---|
| committer | Jerry Ge <jerry.ge@arm.com> | 2024-06-03 15:37:19 +0000 |
| commit | e2f37e91d3328bfaab08b3685dd054f55159ec14 (patch) | |
| tree | d34a1f6e21073fd7f8ba5fb2cdceb69b212f58be /verif | |
| parent | a143d88f37bcf52ad0b796a829542421fe8e1203 (diff) | |
| download | reference_model-main.tar.gz | |
- The existing result_name for constant_net is "result" which
doesn't work for the dynamic_batch_to_space tf test case
- Manually specify the result_name for dynamic_batch_to_space
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: Id028d315d37820f33be1542d7fcacf8a655c5ac8
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: |