diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-04 17:05:24 +0000 |
---|---|---|
committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-30 11:49:56 +0000 |
commit | 4f931307a6319d9d99b3afce4ca6e1cd30d77f01 (patch) | |
tree | 5661b63bd087b210403e3b50dbc0ce0a9f8a41b4 /verif/generator/tosa_arg_gen.py | |
parent | 2d7e4b13d2c3022ae8176d59e2a11d5584ea1d0b (diff) | |
download | reference_model-4f931307a6319d9d99b3afce4ca6e1cd30d77f01.tar.gz |
Main Compliance: DEPTHWISE_CONV2D support
Added DEPTHWISE_CONV2D data generation.
Updated test generation for FP16 and FP32.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I0471d0a1e4e279a27233f4d285082906ceea1bff
Diffstat (limited to 'verif/generator/tosa_arg_gen.py')
-rw-r--r-- | verif/generator/tosa_arg_gen.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py index 4863956..8501caa 100644 --- a/verif/generator/tosa_arg_gen.py +++ b/verif/generator/tosa_arg_gen.py @@ -2038,9 +2038,12 @@ class TosaArgGen: # Compliance - number of dot product calculations if depthwise: - # TODO - add support - dots = 0 + # N*OH*OW*C*M + dots = gtu.product( + (ifm_shape[0], *outputs, *filter_shape[2:]) + ) else: + # N*OH*OW*OC or N*OD*OH*OW*OC dots = gtu.product( (ifm_shape[0], *outputs, filter_shape[0]) ) |