diff --git a/inference-engine/src/transformations/src/transformations/convert_precision.cpp b/inference-engine/src/transformations/src/transformations/convert_precision.cpp index 5feae89287eaa2..ef36b27058d86b 100644 --- a/inference-engine/src/transformations/src/transformations/convert_precision.cpp +++ b/inference-engine/src/transformations/src/transformations/convert_precision.cpp @@ -162,10 +162,11 @@ bool ngraph::pass::ConvertPrecision::run_on_function(std::shared_ptrget_ordered_ops()) { m_transformation_callback(node); - - // Recursively run for TensorIterator body function - if (auto ti = std::dynamic_pointer_cast(node)) { - convert_function_precision(ti->get_body()); + // Recursively apply transformation for sub-graph based operations + if (auto sub_graph_node = std::dynamic_pointer_cast(node)) { + if (auto sub_graph = sub_graph_node->get_function()) { + convert_function_precision(sub_graph); + } } convert_node_input_precision(node); }