We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5549da8 commit 85d6e55Copy full SHA for 85d6e55
torchao/quantization/pt2e/utils.py
@@ -625,8 +625,11 @@ def _is_conv_node(n: Node):
625
"""
626
return n.op == "call_function" and n.target in [
627
torch.ops.aten.conv1d.default,
628
+ torch.ops.aten.conv1d.padding,
629
torch.ops.aten.conv2d.default,
630
+ torch.ops.aten.conv2d.padding,
631
torch.ops.aten.conv3d.default,
632
+ torch.ops.aten.conv3d.padding,
633
]
634
635
0 commit comments