aboutsummaryrefslogtreecommitdiff
path: root/verif/tosa_verif_build_tests.py
diff options
context:
space:
mode:
authorMatthew Haddon <matthew.haddon@arm.com>2021-07-27 09:12:49 +0100
committerEric Kunze <eric.kunze@arm.com>2021-08-19 15:01:20 +0000
commit818ab900ed8e64f43aeebff9924ad883fc349e64 (patch)
treee798900fd1ad317396134668b40a73d4f7c57bdf /verif/tosa_verif_build_tests.py
parenta9017401461224b9bc81e7b1c770ca6091e0e3fb (diff)
downloadreference_model-818ab900ed8e64f43aeebff9924ad883fc349e64.tar.gz
Produce Concat tests with multiple input tensors
* Concat tests now contain between 2 and 5 input tensors concatenated together * Both input and const tensors are used as inputs to the operator * Option to add in const tensor inputs (this is slow), defaults to original behaviour Signed-off-by: Matthew Haddon <matthew.haddon@arm.com> Change-Id: I2a0cc622d31aceab8d24521668d0aae040ba73b1
Diffstat (limited to 'verif/tosa_verif_build_tests.py')
-rwxr-xr-xverif/tosa_verif_build_tests.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/verif/tosa_verif_build_tests.py b/verif/tosa_verif_build_tests.py
index 15482e6..343d8d4 100755
--- a/verif/tosa_verif_build_tests.py
+++ b/verif/tosa_verif_build_tests.py
@@ -192,6 +192,15 @@ def parseArgs():
help="Create test with a particular DType (may be repeated)",
)
+ parser.add_argument(
+ "--num-const-inputs-concat",
+ dest="num_const_inputs_concat",
+ default=0,
+ choices=[0, 1, 2, 3],
+ type=int,
+ help="Allow constant input tensors for concat operator",
+ )
+
args = parser.parse_args()
return args