Reverted conv2d and depthwise_conv2d patterns back

apache · May 24, 2023 · ff1fd37 · ff1fd37
1 parent fe50ae3
commit ff1fd37
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/tvm/relay/op/contrib/ethosu.py b/python/tvm/relay/op/contrib/ethosu.py
@@ -505,9 +505,9 @@ def qnn_conv2d_pattern() -> tvm.relay.dataflow_pattern.DFPattern:
     """
     This function creates the pattern for qnn.conv2D with optional fused RELU activation.
     """
-    pad = is_op("nn.pad")(wildcard(), is_constant())
+    optional_pad = is_op("nn.pad")(wildcard(), is_constant())
     qnn_conv2d = is_op("qnn.conv2d")(
-        pad,
+        optional_pad | wildcard(),
         is_constant(),
         is_constant(),
         is_constant(),
@@ -546,9 +546,9 @@ def qnn_depthwise_conv2d_pattern() -> tvm.relay.dataflow_pattern.DFPattern:
     """
     This function creates the pattern for depthwise qnn.conv2D with optional fused RELU activation.
     """
-    pad = is_op("nn.pad")(wildcard(), is_constant())
+    optional_pad = is_op("nn.pad")(wildcard(), is_constant())
     qnn_conv2d = is_op("qnn.conv2d")(
-        pad,
+        optional_pad | wildcard(),
         is_constant(),
         is_constant(),
         is_constant(),