Extend inputs/outputs info with shape in BA printout (#12453)

This commit is contained in:
Nadezhda Ageeva authored and GitHub committed 2022-08-16 12:21:54 +04:00
1 parent ab3a53a179
commit 3e14988895
1 file changed
+2 -2
@@ -76,7 +76,7 @@ void printInputAndOutputsInfoShort(const T& network) {
}
std::cout << " " << in_name << " (node: " << node_name << ") : " << input.get_element_type() << " / "
<< ov::layout::get_layout(input).to_string() << std::endl;
<< ov::layout::get_layout(input).to_string() << " / " << input.get_partial_shape() << std::endl;
}
std::cout << "Network outputs:" << std::endl;
@@ -105,6 +105,6 @@ void printInputAndOutputsInfoShort(const T& network) {
}
std::cout << " " << out_name << " (node: " << node_name << ") : " << output.get_element_type() << " / "
<< ov::layout::get_layout(output).to_string() << std::endl;
<< ov::layout::get_layout(output).to_string() << " / " << output.get_partial_shape() << std::endl;
}
}