diff options
author | Matthew Haddon <matthew.haddon@arm.com> | 2021-07-27 09:12:49 +0100 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2021-08-19 15:01:20 +0000 |
commit | 818ab900ed8e64f43aeebff9924ad883fc349e64 (patch) | |
tree | e798900fd1ad317396134668b40a73d4f7c57bdf /verif/tosa_verif_build_tests.py | |
parent | a9017401461224b9bc81e7b1c770ca6091e0e3fb (diff) | |
download | reference_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-x | verif/tosa_verif_build_tests.py | 9 |
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 |