diff --git a/python/paddle/static/quantization/quantization_pass.py b/python/paddle/static/quantization/quantization_pass.py
index c9094998dfe2455a6dc2aed2acd86e3f8492cc15..b70386656912199a5f6e1b0b2cfc3d72050fb774 100644
--- a/python/paddle/static/quantization/quantization_pass.py
+++ b/python/paddle/static/quantization/quantization_pass.py
@@ -3226,11 +3226,15 @@ class QuantWeightPass:
                     self._quantized_ops[x_node.name()] = quant_weight_node
 
                 for next_op_node in out_node.outputs:
-                    graph.update_input_link(
-                        out_node,
-                        self._quantized_ops[x_node.name()],
-                        next_op_node,
-                    )
+                    if (
+                        self._quantized_ops[x_node.name()].node
+                        in graph.graph.nodes()
+                    ):
+                        graph.update_input_link(
+                            out_node,
+                            self._quantized_ops[x_node.name()],
+                            next_op_node,
+                        )
                 graph.safe_remove_nodes(_op)
         self._remove_unused_var_nodes(graph)