diff options
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]) ) |