Skip to content

Commit f4705b3

Browse files
committed
Move sorting of initialisers part of OVEP
1 parent 327dbe4 commit f4705b3

File tree

2 files changed

+14
-9
lines changed

2 files changed

+14
-9
lines changed

onnxruntime/core/graph/graph_proto_serializer.cc

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -46,16 +46,10 @@ void GraphViewerToProto(const GraphViewer& graph_view,
4646
std::unordered_set<std::string> current_scope_initializer_set;
4747

4848
auto& initializers = graph_view.GetAllInitializedTensors();
49-
std::vector<std::string> const_inits;
5049
for (auto& it : initializers) {
51-
const_inits.push_back(it.first);
52-
}
53-
std::sort(const_inits.begin(), const_inits.end());
54-
55-
for (auto& it : const_inits){
5650
auto* p_initializer = graph_proto.add_initializer();
57-
*p_initializer = *(initializers.at(it));
58-
current_scope_initializer_set.insert(it);
51+
*p_initializer = *(it.second);
52+
current_scope_initializer_set.insert(it.first);
5953
}
6054

6155

onnxruntime/core/providers/openvino/backend_manager.cc

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,18 @@ BackendManager::GetModelProtoFromFusedNode(const onnxruntime::Node& fused_node,
195195

196196
auto model_proto = model->ToProto();
197197
model_proto->set_ir_version(ONNX_NAMESPACE::Version::IR_VERSION);
198-
subgraph.ToProto(*model_proto->mutable_graph(), true, true);
198+
ONNX_NAMESPACE::GraphProto& graph_proto = *model_proto->mutable_graph();
199+
auto& initializers = subgraph.GetAllInitializedTensors();
200+
std::vector<std::string> const_inits;
201+
for (auto& it : initializers) {
202+
const_inits.push_back(it.first);
203+
}
204+
std::sort(const_inits.begin(), const_inits.end());
205+
206+
for (auto& it : const_inits){
207+
*(graph_proto.mutable_initializer()->Add()) = *(initializers.at(it));
208+
}
209+
subgraph.ToProto(graph_proto, false, true);
199210

200211
#ifndef NDEBUG
201212
if (openvino_ep::backend_utils::IsDebugEnabled()) {

0 commit comments

Comments
 (0)