From c6079ccc113449122de046f9ced8a196808d3844 Mon Sep 17 00:00:00 2001 From: Irina Efode Date: Tue, 11 Jan 2022 17:21:50 +0300 Subject: [PATCH] [IE TESTS] Add body to the reporting (#9430) --- .../src/layer_test_utils/summary.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) 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); }