From 8219518aa8d2a1dc2fe9dec4a24358200e52de4e Mon Sep 17 00:00:00 2001 From: TomWildenhain-Microsoft <67606533+TomWildenhain-Microsoft@users.noreply.github.com> Date: Tue, 6 Apr 2021 22:52:22 -0400 Subject: [PATCH] Fix initializer counts when used as graph output (#7260) Signed-off-by: Tom Wildenhain --- onnxruntime/core/framework/session_state.cc | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/onnxruntime/core/framework/session_state.cc b/onnxruntime/core/framework/session_state.cc index 54a97cb6e4..69cb0377b4 100644 --- a/onnxruntime/core/framework/session_state.cc +++ b/onnxruntime/core/framework/session_state.cc @@ -928,6 +928,12 @@ static void ComputeConstantInitializerUseCount(const Graph& graph, std::unordere } } } + // Initializers can be used as graph outputs + for (const auto* arg : graph.GetOutputs()) { + if (arg->Exists() && graph.GetConstantInitializer(arg->Name(), true /*check_outer_scope*/)) { + constant_initializers_use_count[arg->Name()]++; + } + } } Status SessionState::FinalizeSessionState(const std::basic_string& graph_location,