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

Min. Additional assert in copyTIBoody

Browse Source 
Signed-off-by: Alexander Peskov <alexander.peskov@intel.com>
This commit is contained in:
Alexander Peskov
2020-10-28 01:34:16 +03:00
parent d7e3e92b64
commit fe51b7cb8b
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
inference-engine/src/legacy_api/src/net_pass.cpp
View File

@@ -163,9 +163,17 @@ TensorIterator::Body CopyTIBody(const TensorIterator::Body& body, std::string su
}
TensorIterator::Body res;
for (auto& in : body.inputs) res.inputs.emplace_back(old2new_d[in.get()]);
for (auto& in : body.inputs) {
auto found = old2new_d.find(in.get());
IE_ASSERT(found != old2new_d.end());
res.inputs.emplace_back(found->second);
}
for (auto& out : body.outputs) res.outputs.emplace_back(old2new_d[out.get()]);
for (auto& out : body.outputs) {
auto found = old2new_d.find(out.get());
IE_ASSERT(found != old2new_d.end());
res.outputs.emplace_back(found->second);
}
// Fake holder.
// The graph itself is a shared_ptr set where parent holds child.