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

[HETERO]: Avoid duplicated results (#17725)

Browse Source 
* [HETERO]: Avoid duplicated results

* Code style
This commit is contained in:
Nadezhda Ageeva 2023-05-31 15:13:01 +04:00 committed by GitHub
parent f49935323e
commit 65caa9d745
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/plugins/hetero/executable_network.cpp
View File

@ -285,15 +285,13 @@ HeteroExecutableNetwork::HeteroExecutableNetwork(const InferenceEngine::CNNNetwo
input_subsets[input_subgraph_id].emplace(input); input_subsets[input_subgraph_id].emplace(input);
} }
} }
// for each subset of inputs create separate Result operation if subset belongs to other auto result = std::make_shared<ngraph::op::Result>(output);
result->set_friendly_name(output.get_node()->get_friendly_name() + "_" +
std::to_string(output.get_index()) + "_result");
ngraph::copy_runtime_info(output.get_node_shared_ptr(), result);
subgraphIds.emplace(result.get(), output_subgraph_id);
results.push_back(result);
for (auto&& input_subset : input_subsets) { for (auto&& input_subset : input_subsets) {
auto result = std::make_shared<ngraph::op::Result>(output);
result->set_friendly_name(output.get_node()->get_friendly_name() + "_" +
std::to_string(output.get_index()) + "_" +
std::to_string(input_subset.first) + "_result");
ngraph::copy_runtime_info(output.get_node_shared_ptr(), result);
subgraphIds.emplace(result.get(), output_subgraph_id);
results.push_back(result);
for (auto&& input : input_subset.second) { for (auto&& input : input_subset.second) {
output.remove_target_input(input); output.remove_target_input(input);
auto parameter = auto parameter =