diff options
author | Les Bell <les.bell@arm.com> | 2021-07-28 08:04:55 +0100 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2021-08-12 15:58:58 +0000 |
commit | 2a29dc69170630775523366f29c5914a7981d264 (patch) | |
tree | e8ef987b184a86e92df332742857a6705e36ce5b | |
parent | 33d837e3235d709596cd10d43db18399281b436a (diff) | |
download | reference_model-2a29dc69170630775523366f29c5914a7981d264.tar.gz |
fully_connected test OC independent of input shape
Change-Id: Ib299b78abe0e5c2ef0aeff9d853e583078768e1c
Signed-off-by: Les Bell <les.bell@arm.com>
-rw-r--r-- | verif/tosa_test_gen.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index 99dc5f8..e08add3 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -304,7 +304,13 @@ class TosaTensorGen: assert rank == 2 input_shape = testGen.makeShape(rank) - filter_oc = testGen.makeShape(1)[0] + filter_oc = ( + testGen.rng.integers( + low=testGen.args.tensor_shape_range[0], + high=testGen.args.tensor_shape_range[1], + size=1, + )[0] + ) filter_shape = np.asarray([filter_oc, input_shape[1]]) bias_shape = np.asarray([filter_oc]) |