[LPT] Move clean up transformations in one pass manager #3

eshoguli · Jun 25, 2021 · 6bc05f9 · 6bc05f9
1 parent 7cb423b
commit 6bc05f9
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/inference-engine/src/low_precision_transformations/src/fuse_convert.cpp b/inference-engine/src/low_precision_transformations/src/fuse_convert.cpp
@@ -88,12 +88,14 @@ bool FuseConvertTransformation::transform(TransformationContext& context, ngraph
                     ngraph::op::TemporaryReplaceOutputType(op->get_input_node_shared_ptr(1), element::f32).get());
             NetworkHelper::setOutDataPrecisionForTypeRelaxed(newOp, op->get_output_element_type(0));
             replace_node(op, newOp);
+        } else {
+            return false;
         }
 
-        if (newOp != nullptr) {
-            ngraph::copy_runtime_info({ convert, op }, newOp);
-            newOp->set_friendly_name(op->get_friendly_name());
-        }
+        ngraph::copy_runtime_info({ convert, op }, newOp);
+        newOp->set_friendly_name(op->get_friendly_name());
+
+        register_new_node(newOp);
     }
 
     return true;