From eb2d26f937d6abaae3afa9e95bb88db103e47130 Mon Sep 17 00:00:00 2001
From: Atanas Dimitrov <atanasdimitrov@Atanass-MacBook-Pro.local>
Date: Thu, 14 Dec 2023 15:15:05 +0200
Subject: [PATCH] Call SetOpSchemaFromRegistryForNode while initializing model

---
 onnxruntime/core/graph/graph.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/onnxruntime/core/graph/graph.cc b/onnxruntime/core/graph/graph.cc
index c633895996dda..fd36241c995bf 100644
--- 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_) {