diff --git a/src/tests/ie_test_utils/functional_test_utils/src/layer_test_utils/summary.cpp b/src/tests/ie_test_utils/functional_test_utils/src/layer_test_utils/summary.cpp index 24c5740ed1b..ba4b040085f 100644 --- a/src/tests/ie_test_utils/functional_test_utils/src/layer_test_utils/summary.cpp +++ b/src/tests/ie_test_utils/functional_test_utils/src/layer_test_utils/summary.cpp @@ -163,6 +163,14 @@ void Summary::updateOPsStats(const std::shared_ptr &function, auto loop = ngraph::as_type_ptr(op); auto loop_body = loop->get_function(); updateOPsStats(loop_body, status); + } else if (ngraph::is_type(op)) { + updateOPsStats(op->get_type_info(), status); + auto if_op = ngraph::as_type_ptr(op); + std::vector> 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); }