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

[CPU] Fix case with unused port in Loop body

Browse Source 
Signed-off-by: Alexander Peskov <alexander.peskov@intel.com>
This commit is contained in:
Alexander Peskov 2020-10-29 00:15:53 +03:00
parent 5ac26ff3b6
commit 045d26ff31
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp
View File

@ -132,6 +132,10 @@ void MKLDNNGraph::Replicate(const TensorIterator::Body &subgraph, const MKLDNNEx
graphNodes.push_back(node); graphNodes.push_back(node);
inputNodes[input->getName()] = node; inputNodes[input->getName()] = node;
if (getInputTo(input).empty()) {
unused_data.insert(input);
}
} }
// Step 2. Replicate all internal nodes. // Step 2. Replicate all internal nodes.