diff options
Diffstat (limited to 'verif/generator/tosa_utils.py')
| -rw-r--r-- | verif/generator/tosa_utils.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py index 4a4f6bb..a8e321e 100644 --- a/verif/generator/tosa_utils.py +++ b/verif/generator/tosa_utils.py @@ -10,6 +10,9 @@ from tosa.DType import DType # Maximum dimension size for output and inputs for RESIZE MAX_RESIZE_DIMENSION = 16384 +# Maximum rank of tensor supported by test generator. +MAX_TENSOR_RANK = 6 + # Data type information dictionary # - str: filename abbreviation # - width: number of bytes needed for type |