Merge branch 'main' of https://github.com/microsoft/onnxruntime into …

…qdqmm
microsoft · Dec 25, 2023 · 1c8ae86 · 1c8ae86
2 parents e6c39f4 + 9dd9461
commit 1c8ae86
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/onnxruntime/core/optimizer/pad_fusion.cc b/onnxruntime/core/optimizer/pad_fusion.cc
@@ -48,6 +48,11 @@ bool PadFusion::SatisfyCondition(const Graph& graph, const Node& node, const log
     return false;
   }
 
+  // This pass currently assumed that this attribute already exists on the child node
+  if (child_node.GetAttributes().find("pads") == child_node.GetAttributes().end()) {
+    return false;
+  }
+
   const NodeAttributes& pad_attributes = node.GetAttributes();
   if (pad_attributes.find("mode") != pad_attributes.end() &&
       pad_attributes.at("mode").s() != "constant") {