aboutsummaryrefslogtreecommitdiff
path: root/verif/generator/tosa_utils.py
diff options
context:
space:
mode:
authorJeremy Johnson <jeremy.johnson@arm.com>2024-02-28 17:29:13 +0000
committerJeremy Johnson <jeremy.johnson@arm.com>2024-03-20 09:52:38 +0000
commitdd975b842f5220b57090aacf1845f4ed669182b7 (patch)
tree94a44ef68ce9a8b3db09d7bf8ecba4dda7315cb6 /verif/generator/tosa_utils.py
parentf36f25619cc3a34c75e78637ed244a2ca54ab3f4 (diff)
downloadreference_model-dd975b842f5220b57090aacf1845f4ed669182b7.tar.gz
Fix missing/broken ERROR_IF tests
Fix CONV2D WrongOutputType FP32 & Pad/Stride/DilationSmallerZero issues. Fix PAD WrongInputType. Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com> Change-Id: I57fc57c43e63685e05bf5e3d562c3167411fd57b
Diffstat (limited to 'verif/generator/tosa_utils.py')
-rw-r--r--verif/generator/tosa_utils.py3
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