diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-03-14 16:56:10 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-03-28 15:54:24 +0000 |
commit | 5e36bde3bcfb0c55e61a80cde460ae3955968af3 (patch) | |
tree | 1ee871008a7f1ccd605de174b6aa2d7ba026bb1b /verif/generator/tosa_verif_build_tests.py | |
parent | 08fe7a5b7e2c7c1a77968130e11267ef61490ac8 (diff) | |
download | reference_model-5e36bde3bcfb0c55e61a80cde460ae3955968af3.tar.gz |
Improve CONV2D coverage for conformance
Enable testing of broadcastable bias channel.
Change convolution arguments function to calculate workable
convolutions by adjusting PAD size as needed for FP and errors.
Add --conv-kernel build_tests option to allow different sized
convolution kernels
Improve coverage with batch size, specific shapes and kernels
of 7x7 and 9x9
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I59bb737f398f4363cc606b24fea7812b2059bc01
Diffstat (limited to 'verif/generator/tosa_verif_build_tests.py')
-rw-r--r-- | verif/generator/tosa_verif_build_tests.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/verif/generator/tosa_verif_build_tests.py b/verif/generator/tosa_verif_build_tests.py index 0839cec..f746009 100644 --- a/verif/generator/tosa_verif_build_tests.py +++ b/verif/generator/tosa_verif_build_tests.py @@ -168,6 +168,15 @@ def parseArgs(argv): ) ops_group.add_argument( + "--conv-kernel", + dest="conv_kernels", + action="append", + default=[], + type=lambda x: str_to_list(x), + help="Create convolution tests with a particular kernel shape, e.g., 1,4 or 1,3,1 (only 2D kernel sizes will be used for 2D ops, etc.)", + ) + + ops_group.add_argument( "--max-pooling-padding", dest="max_pooling_padding", default=1, |