diff --git a/backends/cadence/aot/replace_ops.py b/backends/cadence/aot/replace_ops.py index 001ab95d629..c575be6e7fc 100644 --- a/backends/cadence/aot/replace_ops.py +++ b/backends/cadence/aot/replace_ops.py @@ -2250,6 +2250,7 @@ class CommonReplacePasses: ReplaceMMWithAddMMPass, ReplaceRepeatWithCatPass, ReplaceFullLikeWithFullPass, + ReplaceAtenConvolutionWithCadenceConvolutionPass, ] @@ -2282,7 +2283,6 @@ class CadenceReplaceOpsInGraph: RemoveNopSelectOpPass, ReplacePadWithCatPass, ReplaceConstantPadNdWithSlicePass, - ReplaceAtenConvolutionWithCadenceConvolutionPass, ReplaceConvWithChannelLastConvPass, ReplaceTrivialConvWithLinear, ReplaceConvWithIm2RowAndLinear,