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

remove old name tables (#1029)

Browse source 
Remove old hash tables from inference session.
This commit is contained in:
Tracy Sharpe 2019-05-14 21:46:13 -07:00 committed by GitHub
parent 0f6b3ea575
commit a63598ac5d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -791,18 +791,12 @@ common::Status InferenceSession::SaveModelMetadata(const onnxruntime::Model& mod
// save required inputs
const auto& required_inputs = graph.GetInputs(); // inputs excluding initializers
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_model_input_names_.insert(elem->Name());
}
// save all valid inputs
auto& all_inputs = graph.GetInputsIncludingInitializers();
input_def_map_.reserve(all_inputs.size());
model_input_names_.reserve(all_inputs.size());
for (auto elem : all_inputs) {
input_def_map_.insert({elem->Name(), elem});
model_input_names_.insert(elem->Name());
}
// save outputs

20
onnxruntime/core/session/inference_session.h
View file

@ -131,7 +131,7 @@ class InferenceSession {
/**
* Register an execution provider. If you've one to register, call this before invoking Initialize().
* The order of invocation indicates the preference order as well. In other words call this method
* The order of invocation indicates the preference order as well. In other words call this method
* on your most preferred execution provider first followed by the less preferred ones.
* Calling this API is optional in which case onnxruntime will use its internal CPU execution provider.
* @return OK if success.
@ -141,7 +141,7 @@ class InferenceSession {
/**
* Register a graph transformer. If you've one to register, call this before invoking Initialize().
* Calling this API is optional.
* @param[in] - providers Optional. If providers is non-empty this transformer will only to
* @param[in] - providers Optional. If providers is non-empty this transformer will only to
applied to nodes which are assigned to given providers.
* @param[in] - level Optional. Level to which this transformer should be registered. Default is set to 2.
* @return OK if success.
@ -160,9 +160,9 @@ class InferenceSession {
common::Status AddCustomOpDomains(const std::vector<OrtCustomOpDomain*>& ops);
/**
* Register a custom registry for operator schema and kernels. If you've one to register,
* Register a custom registry for operator schema and kernels. If you've one to register,
* call this before invoking Initialize().
* The order of invocation indicates the reversed preference order: Register your most
* The order of invocation indicates the reversed preference order: Register your most
* preferred registry at the end.
* Calling this API is optional.
* @return OK if success.
@ -233,7 +233,7 @@ class InferenceSession {
/**
* Creates a new binding object for binding inputs and outputs.
* @param provider_type specifies the location where the inputs need to be potentially copied.
* @param provider_type specifies the location where the inputs need to be potentially copied.
* See IOBinding class for more info.
*/
common::Status NewIOBinding(std::unique_ptr<IOBinding>* io_binding);
@ -268,9 +268,9 @@ class InferenceSession {
int GetCurrentNumRuns() const;
/**
* Start profiling on this inference session. This simply turns on profiling events to be
* Start profiling on this inference session. This simply turns on profiling events to be
* recorded. A corresponding EndProfiling has to follow to write profiling data to a file.
*@param file_prefix is the prefix of the profile file. It can include a directory path.
*@param file_prefix is the prefix of the profile file. It can include a directory path.
*/
void StartProfiling(const std::string& file_prefix);
#ifdef _WIN32
@ -319,9 +319,7 @@ class InferenceSession {
// Immutable state for each op in the model. Shared by all executors.
SessionState session_state_;
// names of model inputs and outputs used for quick validation.
std::unordered_set<std::string> required_model_input_names_;
std::unordered_set<std::string> model_input_names_;
// names of model outputs used for quick validation.
std::unordered_set<std::string> model_output_names_;
// The file path of where the model was loaded. e.g. /tmp/test_squeezenet/model.onnx
@ -423,7 +421,7 @@ class InferenceSession {
InsertCastTransformer insert_cast_transformer_;
//CustomRegistry objects own the corresponding KernelRegistry and OnnxRuntimeOpSchemaRegistry objects.
//So its lifetime should be same as its constituents. This vector is to extend the lifetime of the owner.
//So its lifetime should be same as its constituents. This vector is to extend the lifetime of the owner.
std::vector<std::shared_ptr<CustomRegistry>> custom_registries_;
};