Call SetOpSchemaFromRegistryForNode while initializing model

microsoft · Dec 14, 2023 · eb2d26f · eb2d26f
1 parent 783bc00
commit eb2d26f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnxruntime/core/graph/graph.cc b/onnxruntime/core/graph/graph.cc
@@ -1310,7 +1310,8 @@ Graph::Graph(const Model& owning_model,
   }
 
   for (const auto& node_proto : graph_proto_->node()) {
-    AddNode(node_proto, name_to_type_map);
+    auto& new_node = AddNode(node_proto, name_to_type_map);
+    SetOpSchemaFromRegistryForNode(new_node);
   }
 
   if (is_loaded_from_model_file_) {