File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
273273class 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
599602class 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
605611class Conv2dReluPattern0 (ConvReluBasePattern ):
You can’t perform that action at this time.
0 commit comments