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

[IE TESTS] Fix bug with constant, const and parameters in summary (#5463)

Browse Source 
* [IE TESTS] Fix bug with constant, const and parameters in summary

* Remove lambda
This commit is contained in:
Irina Efode
2021-04-30 15:48:18 +03:00
committed by GitHub
parent 22e4566faa
commit c99257e1f9
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
inference-engine/tests/ie_test_utils/functional_test_utils/src/layer_test_utils/summary.cpp
View File

@@ -114,25 +114,31 @@ std::map<std::string, PassRate> Summary::getOpStatisticFromReport() {
}
void Summary::updateOPsStats(const std::shared_ptr<ngraph::Function> &function, const PassRate::Statuses &status) {
bool isFunctionalGraph = false;
for (const auto &op : function->get_ordered_ops()) {
if (ngraph::is_type<ngraph::op::Parameter>(op) ||
if (!ngraph::is_type<ngraph::op::Parameter>(op) &&
!ngraph::is_type<ngraph::op::Constant>(op) &&
!ngraph::is_type<ngraph::op::Result>(op)) {
isFunctionalGraph = true;
break;
}
}
for (const auto &op : function->get_ordered_ops()) {
if ((ngraph::is_type<ngraph::op::Parameter>(op) ||
ngraph::is_type<ngraph::op::Constant>(op) ||
ngraph::is_type<ngraph::op::Result>(op)) {
ngraph::is_type<ngraph::op::Result>(op)) && isFunctionalGraph) {
continue;
} else if (ngraph::is_type<ngraph::op::TensorIterator>(op)) {
updateOPsStats(op->get_type_info(), status);
auto ti = ngraph::as_type_ptr<ngraph::op::TensorIterator>(op);
auto ti_body = ti->get_function();
for (const auto &ti_op : ti_body->get_ordered_ops()) {
updateOPsStats(ti_op->get_type_info(), status);
}
updateOPsStats(ti_body, status);
} else if (ngraph::is_type<ngraph::op::v5::Loop>(op)) {
updateOPsStats(op->get_type_info(), status);
auto loop = ngraph::as_type_ptr<ngraph::op::v5::Loop>(op);
auto loop_body = loop->get_function();
for (const auto &loop_op : loop_body->get_ordered_ops()) {
updateOPsStats(loop_op->get_type_info(), status);
}
updateOPsStats(loop_body, status);
} else {
updateOPsStats(op->get_type_info(), status);
}