message: Call (enable|disable)_control_flow_v2 from (enable|disable)_v2_behavior.
diff: @@ -21,6 +21,7 @@ from __future__ import print_function
  from tensorflow.python import tf2
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor_shape
  +from tensorflow.python.ops import control_flow_util
  from tensorflow.python.ops import variable_scope
  from tensorflow.python.util.tf_export import tf_export
  @@ -38,10 +39,12 @@ def enable_v2_behavior():
  This function is called in the main TensorFlow `__init__.py` file, user should
  not need to call it, except during complex migrations.
  """
  - tf2.enable() # Switches TensorArrayV2 and control flow V2
  + tf2.enable()
  ops.enable_eager_execution()
  tensor_shape.enable_v2_tensorshape() # Also switched by tf2
  variable_scope.enable_resource_variables()
  + # Enables TensorArrayV2 and control flow V2.
  + control_flow_util.enable_control_flow_v2()
  @tf_export(v1=["disable_v2_behavior"])
  @@ -55,7 +58,9 @@ def disable_v2_behavior():
  User can call this function to disable 2.x behavior during complex migrations.
  """
  - tf2.disable() # Switches TensorArrayV2 and control flow V2
  + tf2.disable()
  ops.disable_eager_execution()
  tensor_shape.disable_v2_tensorshape() # Also switched by tf2
  variable_scope.disable_resource_variables()
  + # Disables TensorArrayV2 and control flow V2.
  + control_flow_util.disable_control_flow_v2()
  