commit e2325d12ba6eebeb59d50e7a7ce578a8a32a03ed
author Won Jeon <won.jeon@arm.com> Sat Jun 10 15:25:54 2023 +0000
committer Won Jeon <won.jeon@arm.com> Wed Jul 05 21:20:18 2023 +0000
tree 24bed603251afc3ecc8b0fd5e1793a6278dda39a
parent edac6abb626993c1005feb48db497064a7d102d0

Add a parameter to tensor generation function to disable fuzzing

Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Iff31b33b818a181371904915d5477a169513aa2e

verif/frameworks/tensor_gen.py

diff --git a/verif/frameworks/tensor_gen.py b/verif/frameworks/tensor_gen.py
index 170e5d8..d50bc74 100644
--- a/verif/frameworks/tensor_gen.py
+++ b/verif/frameworks/tensor_gen.py
@@ -91,7 +91,7 @@
         return tf_placeholders, tf_consts
 
     @staticmethod
-    def tgBFuzz(op, shape, dtype, rng):
+    def tgBFuzz(op, shape, dtype, rng, fuzzed=[]):
         # Build random tensor placeholder node args of a given shape, optionally
         # fuzzing the arguments with random 1's to force broadcasting
 
@@ -105,12 +105,14 @@
         tf_placeholders = []
         tf_consts = []
         for i in range(pl):
-            if i == fuzz_arg:
+            if not fuzzed and i == fuzz_arg:
                 # Insert the broadcast in one dimension index
                 s_fuzz = list(shape)
                 s_fuzz[fuzz_idx] = 1
                 s_fuzz = tuple(s_fuzz)
                 i_shape = s_fuzz
+                # Record the fuzzed index.
+                fuzzed.append(i)
             else:
                 i_shape = shape
             tf_placeholders.append(