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

Copy inputs and outputs directly in InferenceSession::SaveModelMetadata (#777)

Browse source 
* Copy required inputs and outputs directly in InferenceSession::SaveModelMetadata

* trivial

* trivial
This commit is contained in:
Tao Qin 2019-04-05 15:16:55 -07:00 committed by GitHub
parent 3eddb2d61e
commit 58ef1306d4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onnxruntime/core/session/inference_session.cc
View file

@ -220,7 +220,7 @@ common::Status InferenceSession::RegisterExecutionProvider(std::unique_ptr<IExec
return Status::OK();
}
common::Status InferenceSession::RegisterGraphTransformer(std::unique_ptr<onnxruntime::GraphTransformer> p_graph_transformer,
common::Status InferenceSession::RegisterGraphTransformer(std::unique_ptr<onnxruntime::GraphTransformer> p_graph_transformer,
TransformerLevel level) {
if (p_graph_transformer == nullptr) {
return Status(common::ONNXRUNTIME, common::FAIL, "Received nullptr for graph transformer");
@ -932,10 +932,9 @@ common::Status InferenceSession::SaveModelMetadata(const onnxruntime::Model& mod
// save required inputs
const auto& required_inputs = graph.GetInputs(); // inputs excluding initializers
required_input_def_list_.reserve(required_inputs.size());
required_input_def_list_ = required_inputs; // A direct copy of required inputs
required_model_input_names_.reserve(required_inputs.size());
for (const auto& elem : required_inputs) {
required_input_def_list_.push_back(elem);
required_model_input_names_.insert(elem->Name());
}
@ -950,10 +949,9 @@ common::Status InferenceSession::SaveModelMetadata(const onnxruntime::Model& mod
// save outputs
const auto& outputs = graph.GetOutputs();
output_def_list_.reserve(outputs.size());
output_def_list_ = outputs; // A direct copy of outputs
model_output_names_.reserve(outputs.size());
for (const auto& elem : outputs) {
output_def_list_.push_back(elem);
model_output_names_.insert(elem->Name());
}