We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4b0f3f6 commit fac6e34Copy full SHA for fac6e34
1 file changed
src/moduleprofiler/ops.py
@@ -602,11 +602,7 @@ def _batchnorm1d_ops_fn(
602
total_ops = 5 * num_elements + 4
603
604
else:
605
- if module.bias is not None:
606
- total_ops = 7 * num_elements + 4
607
-
608
- else:
609
- total_ops = 6 * num_elements + 4
+ total_ops = 7 * num_elements + 4
610
611
# Add num_features C
612
total_ops *= module.num_features
@@ -625,11 +621,7 @@ def _batchnorm2d_ops_fn(
625
621
626
622
627
623
628
629
630
631
632
624
633
634
635
0 commit comments