Skip to content

Commit 3d63ad8

Browse files
authored
Add quantizer support for quantized_conv1d_ncl (#17938)
Differential Revision: D95279330 Pull Request resolved: #17938
1 parent 520566c commit 3d63ad8

3 files changed

Lines changed: 536 additions & 124 deletions

File tree

backends/cadence/aot/quantizer/patterns.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -267,7 +267,7 @@ def get_anchors(
267267
)
268268

269269
def replacement_op(self) -> OpOverload:
270-
return torch.ops.cadence.quantized_conv2d_nchw.per_tensor
270+
return torch.ops.cadence.quantized_conv1d_ncl.per_tensor
271271

272272

273273
class Conv2dPattern(QuantizationPattern):
@@ -594,12 +594,18 @@ class Conv1dReluPattern0(ConvReluBasePattern):
594594
def partition_types(self) -> List[OpOverload]:
595595
return [torch.ops.aten.conv1d.default, torch.ops.aten.relu.default]
596596

597+
def replacement_op(self) -> OpOverload:
598+
return torch.ops.cadence.quantized_conv1d_ncl.per_tensor
599+
597600

598601
# Conv1d + alternate relu op fusion
599602
class Conv1dReluPattern1(ConvReluBasePattern):
600603
def partition_types(self) -> List[OpOverload]:
601604
return [torch.ops.aten.conv1d.default, torch.ops.aten.relu_.default]
602605

606+
def replacement_op(self) -> OpOverload:
607+
return torch.ops.cadence.quantized_conv1d_ncl.per_tensor
608+
603609

604610
# Conv2d + regular relu op fusion
605611
class Conv2dReluPattern0(ConvReluBasePattern):

0 commit comments

Comments
 (0)