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,