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

Small addendum to #9624 (#9639)

Browse source
This commit is contained in:
Hariharan Seshadri 2021-11-01 15:54:21 -07:00 committed by GitHub
parent 1240c595a9
commit f41396eff0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -688,6 +688,19 @@ void SessionState::ResolveMemoryPatternFlag() {
break;
}
}
// For subgraphs, the implicit inputs need to meet the same crieria
// as the explicit inputs for memory pattern to be enabled
if (graph_viewer_->IsSubgraph()) {
const auto* parent_node = graph_viewer_->ParentNode();
for (auto* implicit_input : parent_node->ImplicitInputDefs()) {
if (!implicit_input->HasTensorOrScalarShape()) {
enable_mem_pattern_ = false;
break;
}
}
}
}
}