diff options
author | Matthew Haddon <matthew.haddon@arm.com> | 2021-08-20 15:02:39 +0100 |
---|---|---|
committer | Matthew Haddon <matthew.haddon@arm.com> | 2021-08-31 08:11:43 +0100 |
commit | 82ad4d628cdb2e20975d8b7b43b41bbf381b015f (patch) | |
tree | 2598b54ce55b6e6dca0f77f72b5894192fa18356 | |
parent | a9d539548ab90693682683b4c685a1597b1c8c26 (diff) | |
download | reference_model-82ad4d628cdb2e20975d8b7b43b41bbf381b015f.tar.gz |
Set TILE multiple to one if input dimension is large
* Sets multiple of input dimension to one if the dimension of the
input tensor is large to reduce ouput tensor size.
Change-Id: I2ebcecf438282de032a33e0cf4b3847cbf94440d
-rw-r--r-- | verif/tosa_test_gen.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index efc819c..d41d6e2 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -784,8 +784,13 @@ class TosaArgGen: # enormous tensors multiples = [] for i in range(rank): - multiples.append(testGen.randInt(1, 4)) - + if ifm_shape[i] > 1000: + # Multiple of 1 if ifm_shape dimension is large to reduce tensor size + multiples.append(1) + elif max(ifm_shape) > 1000: + multiples.append(2) + else: + multiples.append(testGen.randInt(1, 4)) arg_list.append(("perm{}".format(p), [multiples])) return arg_list |