diff options
author | James Ward <james.ward@arm.com> | 2023-02-02 14:56:33 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-02-22 16:28:39 +0000 |
commit | 30124a8d9ab659f2661f3bb5962ae074d9ac4fa3 (patch) | |
tree | 305d31c047dd4704f8496e37026bd40a7b4c9f99 /verif/generator/tosa_test_gen.py | |
parent | c5d759328fbfe9772fba56e63c81d196b3a7620f (diff) | |
download | reference_model-30124a8d9ab659f2661f3bb5962ae074d9ac4fa3.tar.gz |
Bug fixes for max-batch-size/ofm-depth
* Conformance regen required for tosa-bi: conv2d conv3d depthwise_conv2d transpose_conv2d
* Include bug fix for testGen.makeShape()
* Include json changes to limit size of tests which now have ofm_depth > 1
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: Ic8221b9a742f5737df523997bee7260f4dfcdef3
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index b48810f..a768da0 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -219,6 +219,11 @@ class TosaTestGen: shape[0] = min(shape[0], self.args.max_batch_size) return shape + def makeDimension(self): + return self.randInt( + low=self.args.tensor_shape_range[0], high=self.args.tensor_shape_range[1] + ) + # Argument generators # Returns a list of tuples (stringDescriptor, [build_fcn_arg_list]) # Where the string descriptor is used to generate the test name and |