Skip to content

Commit 064671b

Browse files
committed
Update comments
1 parent 3c7fe32 commit 064671b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backends/xnnpack/quantizer/xnnpack_quantizer_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -331,6 +331,8 @@ def _do_annotate_conv(
331331
):
332332
weight_val = weight.meta.get("val", None)
333333
weight_shape = getattr(weight_val, "shape", None)
334+
335+
# Skip if not a 4D weight tensor (i.e. not conv2d)
334336
if weight_shape is not None and len(weight_shape) != 4:
335337
continue
336338

0 commit comments

Comments
 (0)