code: def _control_flow_v2():
      enable_control_flow_v2_old = (
          control_flow_util.ENABLE_CONTROL_FLOW_V2
      )
      control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
      try:
          yield
      finally:
          control_flow_util.ENABLE_CONTROL_FLOW_V2 = (
              enable_control_flow_v2_old
          )
  
imports: import collections
  import contextlib
  import six
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.auto_batching import instructions
  from tensorflow_probability.python.experimental.auto_batching import xla
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow.python.ops import control_flow_util
label_desc: Use image node_id 1 for calling a global function with example usage: _control_flow_v2() without return types