From fa8768723ac690fffa4d2d457c6d96fe55906ae1 Mon Sep 17 00:00:00 2001 From: Ankur Verma <31362771+ankurverma85@users.noreply.github.com> Date: Tue, 29 Jun 2021 16:54:36 -0700 Subject: [PATCH] Allow custom loaders for testing (#8150) --- onnxruntime/core/session/inference_session.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/session/inference_session.h b/onnxruntime/core/session/inference_session.h index 6033b72731..cfc45b120f 100644 --- a/onnxruntime/core/session/inference_session.h +++ b/onnxruntime/core/session/inference_session.h @@ -452,6 +452,9 @@ class InferenceSession { */ common::Status Load(std::unique_ptr p_model_proto) ORT_MUST_USE_RESULT; + common::Status Load(std::function&)> 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&)> loader, - const std::string& event_name) ORT_MUST_USE_RESULT; template common::Status Load(const std::basic_string& model_uri) ORT_MUST_USE_RESULT;