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

Removed WA with adding fake output for VariadicSplit operation (#1938)

Browse Source
This commit is contained in:
Gleb Kazantaev
2020-08-26 06:28:03 +03:00
committed by GitHub
parent c4920ef5a0
commit 12197a4800
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
inference-engine/src/transformations/src/transformations/optimize_strided_slice.cpp
View File

@@ -194,10 +194,7 @@ bool ngraph::pass::GroupedStridedSliceOptimizer::run_on_function(std::shared_ptr
auto i = 0;
NodeVector ops_to_replace;
for (auto & record : output_to_size) {
if (record.first == fake_output) {
auto& results = const_cast<::ngraph::ResultVector&>(f->get_results());
results.push_back(std::make_shared<ngraph::opset1::Result>(variadic_split->output(i)));
} else {
if (record.first != fake_output) {
record.first.replace(variadic_split->output(i));
ops_to_replace.push_back(record.first.get_node_shared_ptr());
}