diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-25 12:53:21 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2024-01-30 16:11:26 +0000 |
| commit | fc4bde92120567a98189f95cfe90bb1699d25809 (patch) | |
| tree | 48ab320c288440816734fc771a817257f3a229f9 /verif/generator/tosa_utils.py | |
| parent | 95a6710ffb8cadcb8658a967ab29cac1bffad930 (diff) | |
| download | reference_model-fc4bde92120567a98189f95cfe90bb1699d25809.tar.gz | |
Fix up shape operator test errors
Update serialization_lib to store SHAPE as INT64.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ie589cd6670dc79b77df981c81cd7c27b982f20fa
Diffstat (limited to 'verif/generator/tosa_utils.py')
| -rw-r--r-- | verif/generator/tosa_utils.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py index 33db95f..6387d06 100644 --- a/verif/generator/tosa_utils.py +++ b/verif/generator/tosa_utils.py @@ -56,7 +56,7 @@ def dtypeIsSupportedByCompliance(dtype): """Types supported by the new data generation and compliance flow.""" if isinstance(dtype, list) or isinstance(dtype, tuple): dtype = dtype[0] - return dtype in (DType.FP32, DType.FP16, DType.SHAPE) + return dtype in (DType.FP32, DType.FP16) def getOpNameFromOpListName(opName): |