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

Allow custom loaders for testing (#8150)

Browse source
This commit is contained in:
Ankur Verma 2021-06-29 16:54:36 -07:00 committed by GitHub
parent 507d97b200
commit fa8768723a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -452,6 +452,9 @@ class InferenceSession {
*/
common::Status Load(std::unique_ptr<ONNX_NAMESPACE::ModelProto> p_model_proto) ORT_MUST_USE_RESULT;
common::Status Load(std::function<common::Status(std::shared_ptr<Model>&)> loader,
const std::string& event_name) ORT_MUST_USE_RESULT;
common::Status DoPostLoadProcessing(onnxruntime::Model& model) ORT_MUST_USE_RESULT;
#endif // !defined(ORT_MINIMAL_BUILD)
@ -497,8 +500,6 @@ class InferenceSession {
common::Status SaveModelMetadata(const onnxruntime::Model& model) ORT_MUST_USE_RESULT;
#if !defined(ORT_MINIMAL_BUILD)
common::Status Load(std::function<common::Status(std::shared_ptr<Model>&)> loader,
const std::string& event_name) ORT_MUST_USE_RESULT;
template <typename T>
common::Status Load(const std::basic_string<T>& model_uri) ORT_MUST_USE_RESULT;