diff --git a/tf_adapter/python/npu_bridge/__init__.py b/tf_adapter/python/npu_bridge/__init__.py
index 428dd33a067b352d2f35b5107dea1ac97cfd9137..302936a902ebd4c7061db491cafe4d5918d7160f 100644
--- a/tf_adapter/python/npu_bridge/__init__.py
+++ b/tf_adapter/python/npu_bridge/__init__.py
@@ -17,6 +17,11 @@
 
 
 import atexit
+import tensorflow
+
+print("Enable tensorflow control flow v2 and resource variables", flush=True)
+tensorflow.enable_control_flow_v2()
+tensorflow.enable_resource_variables()
 from npu_bridge.helper.helper import npu_bridge_handle
 from npu_bridge.helper.helper import version as __version__
 from npu_bridge.helper import helper