diff options
Diffstat (limited to 'verif/generator/tosa_utils.py')
-rw-r--r-- | verif/generator/tosa_utils.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py index 3d733f4..33db95f 100644 --- a/verif/generator/tosa_utils.py +++ b/verif/generator/tosa_utils.py @@ -1,4 +1,4 @@ -# Copyright (c) 2021-2023, ARM Limited. +# Copyright (c) 2021-2024, ARM Limited. # SPDX-License-Identifier: Apache-2.0 import struct import sys @@ -49,13 +49,14 @@ class DataGenType(IntEnum): OP_BOUNDARY = 2 OP_FULLSET = 3 OP_SPECIAL = 4 + FIXED_DATA = 5 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) + return dtype in (DType.FP32, DType.FP16, DType.SHAPE) def getOpNameFromOpListName(opName): |