saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-17 21:10:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix initializer counts when used as graph output (#7260)

Browse source 
Signed-off-by: Tom Wildenhain <tomwi@microsoft.com>
This commit is contained in:
TomWildenhain-Microsoft 2021-04-06 22:52:22 -04:00 committed by GitHub
parent 2ec452cdad
commit 8219518aa8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/core/framework/session_state.cc
View file

@ -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<PATH_CHAR_TYPE>& graph_location,