saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-29 23:06:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Put the initializers at the end of the cluster inputs list (#1751)

Browse source 
Restore the missing variable
This commit is contained in:
Tomasz Dołbniak 2019-09-04 00:09:37 +02:00 committed by jywu-msft
parent 9523977cc2
commit 4ed8d4b30e
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
onnxruntime/core/providers/ngraph/ngraph_execution_provider.cc
View file

@ -438,15 +438,26 @@ static void GetInputsOutputsOfCluster(const GraphViewer& graph_viewer,
}
const auto& initializers = graph_viewer.GetAllInitializedTensors();
std::vector<std::string> const_inputs;
for (const auto& in_arg : ordered_input_args) {
if ((initializers.count(in_arg) && !original_graph_inputs.count(in_arg)) ||
ng_required_initializers.count(in_arg)) {
cluster_inputs.push_back(in_arg);
} else if (!output_args.count(in_arg)) {
const_inputs.push_back(in_arg);
}
}
for (const auto& in_arg : ordered_input_args) {
if (!output_args.count(in_arg) &&
!((initializers.count(in_arg) && !original_graph_inputs.count(in_arg)) ||
ng_required_initializers.count(in_arg))) {
cluster_inputs.push_back(in_arg);
}
}
for (const auto& in_arg : const_inputs) {
cluster_inputs.push_back(in_arg);
}
std::copy(external_output_args.begin(), external_output_args.end(), std::back_inserter(cluster_outputs));
for (const auto& node_arg : graph_viewer.GetOutputs()) {
const auto& name = node_arg->Name();