diff options
| author | Won Jeon <won.jeon@arm.com> | 2023-09-18 16:32:45 -0700 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2023-09-28 18:26:39 +0000 |
| commit | f9c0ceea99e197ab14f779eb51c5e1479dbeb4dd (patch) | |
| tree | 9484444d75f38c533c214c6568671437b7fbddf4 /verif/frameworks/test_gen_utils.py | |
| parent | 41ebe72588b20b912eb8c9e082b2d66b37564ad3 (diff) | |
| download | reference_model-f9c0ceea99e197ab14f779eb51c5e1479dbeb4dd.tar.gz | |
Add 0-rank tensor support for concat in framework test
Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Iff77091e4a57f487431ffbf7ac1c89301a153c8b
Diffstat (limited to 'verif/frameworks/test_gen_utils.py')
| -rw-r--r-- | verif/frameworks/test_gen_utils.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/verif/frameworks/test_gen_utils.py b/verif/frameworks/test_gen_utils.py index 6a59848..f31ac63 100644 --- a/verif/frameworks/test_gen_utils.py +++ b/verif/frameworks/test_gen_utils.py @@ -9,6 +9,9 @@ import tensorflow as tf # Get a string name for a given shape def get_shape_str(shape, dtype): shape_name = None + if len(shape) == 0: + shape_name = "0" + for dim in shape: shape_name = (shape_name + "x" + str(dim)) if shape_name else str(dim) |