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

Fixed KW hits (#4638)

Browse Source
This commit is contained in:
Ilya Churaev 2021-03-05 11:13:11 +03:00 committed by GitHub
parent 3cd5da0797
commit 4675a12c8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ngraph/core/reference/src/runtime/reference/loop.cpp
View File

@ -194,8 +194,10 @@ namespace ngraph
// Check execution condition
bool body_exec_condition(false);
body_outputs[special_ports.body_condition_output_idx]->read(
&body_exec_condition, sizeof(bool));
if (body_outputs.size() > special_ports.body_condition_output_idx &&
body_outputs[special_ports.body_condition_output_idx])
body_outputs[special_ports.body_condition_output_idx]->read(
&body_exec_condition, sizeof(bool));
if (!body_exec_condition)
break;

2
ngraph/core/reference/src/runtime/reference/tensor_iterator.cpp
View File

@ -161,6 +161,8 @@ namespace ngraph
for (size_t i = 0; i < concat_outputs.size(); ++i)
{
const auto& concat_desc = concat_outputs[i];
if (!concat_desc)
continue;
auto shape =
func->get_results().at(concat_desc->m_body_value_index)->get_shape();
std::vector<Shape> shapes_to_concat(values_to_concat[i].size(), shape);