natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE Myriad] - WrapInLoop fix: if data has consumer's input inside subgraph - replace them (#958)

This commit is contained in:
Daria Mityagina 2020-06-17 17:27:17 +03:00 committed by GitHub
parent 14b44803ba
commit dea04dae8c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
inference-engine/src/vpu/graph_transformer/src/middleend/passes/adjust_data_batch.cpp
View File

@ -193,10 +193,10 @@ void PassImpl::wrapInLoop(const Model& model, const StageList& subgraph) {
loopEndOutputs.push_back(originalOutput);
const auto rule = IterationRule{Dim::N, 0, 1, -1};
endIterationComponents.emplace(std::make_pair(loopEndOutputs.size() - 1, rule), loopEndInputs.size() - 1);
} else {
for (const auto& consumerEdge : originalOutput->consumerEdges()) {
}
for (const auto& consumerEdge : originalOutput->consumerEdges()) {
if (subgraph.has(consumerEdge->consumer()))
model->replaceStageInput(consumerEdge, output);
}
}
}
}