Update rank limits for SLICE, TILE and TRANSPOSE

Updated to align with corresponding changes to the
spec.

In addition, some ERROR_IF tests have been updated to
match the checks specified by the spec, including:
PAD, SLICE, TILE, TRANSPOSE.

Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: Ie2c5f48e79a5610eb82739170e25057a63dac1d8

1 files changed, 11 insertions, 0 deletions

diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py
index 29ae898..8ff62f1 100644
--- a/verif/generator/tosa_utils.py
+++ b/verif/generator/tosa_utils.py
@@ -148,6 +148,17 @@ def get_wrong_output_type(op_name, rng, input_dtype):
     return rng.choice(a=incorrect_types)
 
 
+def get_rank_mismatch_shape(rng, output_shape):
+    """
+    Extends the rank of the provided output_shape by
+    an arbitrary amount but ensures the total element
+    count remains the same.
+    """
+    rank_modifier = rng.choice([1, 2, 3])
+    output_shape += [1] * rank_modifier
+    return output_shape
+
+
 def float32_is_valid_bfloat16(f):
     """Return True if float value is valid bfloat16."""
     f32_bits = get_float32_bitstring(f)