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

Fix redundant population of output_indices_ in the execution frame and reserve memory for it in advance. (#375)

Browse source
This commit is contained in:
Pranav Sharma 2019-01-23 21:19:31 -08:00 committed by GitHub
parent 82b4ec3d42
commit 61bbf4bfcc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
onnxruntime/core/framework/execution_frame.cc
View file

@ -338,20 +338,14 @@ void ExecutionFrame::Init(const onnxruntime::GraphViewer& graph,
}
// 4. Handle non-empty output vector
// setup output_indices_, we dont' want to generate mem plan on output tensors.
for (const auto& oname : output_names) {
int mlvalue_idx;
Status status = mlvalue_idx_map.GetIdx(oname, mlvalue_idx);
ORT_ENFORCE(status.IsOK(), status.ErrorMessage());
output_indices_.push_back(mlvalue_idx);
}
if (!fetches.empty()) {
// should've already verified this much before when Run() starts
ORT_ENFORCE(output_names.size() == fetches.size(),
"output_names vector size: " + std::to_string(output_names.size()) +
" does not match that of fetches vector: " + std::to_string(fetches.size()));
// setup output_indices_, we dont' want to generate mem plan on output tensors.
output_indices_.reserve(output_names.size());
auto idx = 0;
for (const auto& oname : output_names) {
int mlvalue_idx;