[opt:fuse_pad_into_conv] remove auto_pad, when use pads

Signed-off-by: 462630221 <[email protected]>
onnx · Feb 20, 2021 · 8f5a6e9 · 8f5a6e9
1 parent c40d3c0
commit 8f5a6e9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/onnxoptimizer/passes/fuse_pad_into_conv.h b/onnxoptimizer/passes/fuse_pad_into_conv.h
@@ -165,6 +165,11 @@ struct FusePadIntoConv final : public PredicateBasedPass {
       conv_pads[conv_pads_size / 2 + j] += pads[pads_size / 2 + i];
     }
 
+    // Clean the auto_pad
+    if (conv->hasAttribute(Symbol("auto_pad"))){
+      conv->removeAttribute(Symbol("auto_pad"));
+    }
+
     conv->is_(kpads, std::move(conv_pads));
     conv->replaceInput(0, pad->inputs()[0]);
     pad->destroy();