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

[IE TESTS] Add body to the reporting (#9430)

Browse Source
This commit is contained in:
Irina Efode 2022-01-11 17:21:50 +03:00 committed by GitHub
parent f96a5183fb
commit c6079ccc11
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/tests/ie_test_utils/functional_test_utils/src/layer_test_utils/summary.cpp
View File

@ -163,6 +163,14 @@ void Summary::updateOPsStats(const std::shared_ptr<ngraph::Function> &function,
auto loop = ngraph::as_type_ptr<ngraph::op::v5::Loop>(op);
auto loop_body = loop->get_function();
updateOPsStats(loop_body, status);
} else if (ngraph::is_type<ngraph::op::v8::If>(op)) {
updateOPsStats(op->get_type_info(), status);
auto if_op = ngraph::as_type_ptr<ngraph::op::v8::If>(op);
std::vector<std::shared_ptr<ngraph::Function>> bodies;
for (size_t i = 0; i < if_op->get_internal_subgraphs_size(); i++) {
auto if_body = if_op->get_function(i);
updateOPsStats(if_body, status);
}
} else {
updateOPsStats(op->get_type_info(), status);
}