saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-06 00:03:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[VitisAI] graph_save only saves proto of the graph instead of entire model (#17368)

Browse source 
### Description
<!-- Describe your changes. -->
graph_save only saves proto of the graph instead of entire model.


### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
We would like to export a part of a model as a new model for unit test.
Therefore, we have to change the API to support such need.
This commit is contained in:
BoarQing 2023-09-06 14:40:48 -05:00 committed by GitHub
parent 8914fe687b
commit 2629cb8606
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/core/providers/vitisai/imp/graph.cc
View file

@ -151,8 +151,10 @@ void graph_save(const Graph& graph, const std::string& filename, const std::stri
}
}
// use relative path as data storage.
for (auto i = 0; i < model_proto.graph().initializer_size(); ++i) {
auto initializer = model_proto.mutable_graph()->mutable_initializer(i);
auto graph_proto = model_proto.mutable_graph();
*graph_proto = graph.ToGraphProto();
for (auto i = 0; i < graph_proto->initializer_size(); ++i) {
auto initializer = graph_proto->mutable_initializer(i);
for (auto j = 0; j < initializer->external_data_size(); ++j) {
auto external_data = initializer->mutable_external_data(j);
if (external_data->key() == "location") {