code: def CompiledKernel(fn, *inputs, **kwargs):
      """Execute 'fn' as a compiled XLA kernel, with 'inputs'."""
      name = kwargs.pop("name", None)
      noinline = kwargs.pop("noinline", None)
  
      @function.Defun(func_name=name, noinline=noinline, compiled=True)
      def Compiled(*args):
          return fn(*args)
  
      return Compiled(*inputs)
  
imports: import os
  import numpy
  from tensorflow.compiler.tests import test_utils
  from tensorflow.core.protobuf import config_pb2
  from tensorflow.core.protobuf import rewriter_config_pb2
  from tensorflow.python.client import session
  from tensorflow.python.compiler.xla import jit
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import function
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import cond
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import while_loop
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling a global function with example usage: CompiledKernel(fn) and returns: Compiled, fn