From 54bb61effee583239d30ec6d4fda32c1a710050c Mon Sep 17 00:00:00 2001 From: Jerry Ge Date: Wed, 20 Dec 2023 22:21:24 +0000 Subject: Fix dynamic_input_shape referenced before assignment error - Found the error for framework test generator when it's not generating dynamic_shaped tests Signed-off-by: Jerry Ge Change-Id: I54bf1422026f7f962d3f6ed0be9bd2e4dade14c5 --- verif/frameworks/tosa_verif_framework_generator.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 1b187ba..3a9c0ca 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -1,5 +1,5 @@ #!/usr/bin/env python3 -# Copyright (c) 2020-2023, ARM Limited. +# Copyright (c) 2020-2024, ARM Limited. # SPDX-License-Identifier: Apache-2.0 import argparse import os @@ -1168,22 +1168,22 @@ def run_unit_test( placeholder_shapes = [] for idx, (name, val) in enumerate(placeholders): - input_shape = val.shape + input_shape = tuple(val.shape) + try: dynamic_shape_dim_tuples = op["dynamic_shape_dim"] dim_tuple = dynamic_shape_dim_tuples[idx] dim = dim_tuple[0] - input_shape = list(val.shape) + input_shape = list(input_shape) input_shape[dim] = None - dynamic_input_shape = tuple(input_shape) - addl_args.append(dynamic_input_shape) + addl_args.append(tuple(input_shape)) except KeyError: pass placeholder_names.append(name) placeholder_signatures = placeholder_signatures + ( - tf.TensorSpec(shape=dynamic_input_shape, dtype=val.dtype, name=name), + tf.TensorSpec(shape=input_shape, dtype=val.dtype, name=name), ) placeholder_npy_filenames.append("{}.npy".format(name.split(":")[0])) placeholder_shapes.append(val.shape) -- cgit v1.2.1