code: def set_tf_xla_forced_compile(b):
      global _tf_xla_forced_compile_enabled
      _tf_xla_forced_compile_enabled = b
  
file_path: trax/trax/fastmath/tf.py
imports: import numpy
  import tensorflow.compat.v2
  from trax.shapes import ShapeDtype
  from trax.tf_numpy import extensions
  from trax.tf_numpy import numpy
label_desc: Use image node_id 4 for calling a global function with example usage: set_tf_xla_forced_compile(b) without return types