diff options
Diffstat (limited to 'verif/frameworks/test_gen_utils.py')
-rw-r--r-- | verif/frameworks/test_gen_utils.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/verif/frameworks/test_gen_utils.py b/verif/frameworks/test_gen_utils.py index 2d8e5d6..6a59848 100644 --- a/verif/frameworks/test_gen_utils.py +++ b/verif/frameworks/test_gen_utils.py @@ -30,6 +30,8 @@ def get_shape_str(shape, dtype): shape_name = shape_name + "_qi16" elif dtype == tf.quint16: shape_name = shape_name + "_qu16" + elif dtype == tf.complex64: + shape_name = shape_name + "_c64" else: raise Exception("Unsupported type: {}".format(dtype)) |