code: def set_config_v2(enable_xla=False):
      """Config eager context according to flag values using TF 2.0 API."""
      if enable_xla:
          tf.config.optimizer.set_jit(True)
          # Disable PinToHostOptimizer in grappler when enabling XLA because it
          # causes OOM and performance regression.
          tf.config.optimizer.set_experimental_options(
              {"pin_to_host_optimization": False}
          )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import time
  import tensorflow
  from tensorflow.core.protobuf import rewriter_config_pb2
  from tensorflow.python import tf2
  from tensorflow.python.eager import profiler
label_desc: Use image node_id 4 for calling a global function with example usage: set_config_v2(enable_xla) without return types