diff --git a/onnxruntime/core/session/inference_session.cc b/onnxruntime/core/session/inference_session.cc index 2a155eb9d4..3cc4455618 100644 --- a/onnxruntime/core/session/inference_session.cc +++ b/onnxruntime/core/session/inference_session.cc @@ -220,7 +220,7 @@ common::Status InferenceSession::RegisterExecutionProvider(std::unique_ptr p_graph_transformer, +common::Status InferenceSession::RegisterGraphTransformer(std::unique_ptr 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()); }