docstr: Tests that xla.compile works in tf.function.
code: def test_xla_compile_in_function(self):
      """Tests that xla.compile works in tf.function."""
  
      @def_function.function
      def func_wrapper(a):
          def compute(a):
              return a + 1
  
          return xla.compile(compute, [a])
  
      self.assertEqual(self.evaluate(func_wrapper(1))[0], 2)
  
imports: from absl.testing import parameterized
  from tensorflow.python import summary
  from tensorflow.python.compiler.xla import xla
  from tensorflow.python.eager import def_function
  from tensorflow.python.estimator import model_fn
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.ops import control_flow_util
  from tensorflow.python.ops import logging_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import state_ops
  from tensorflow.python.ops import variable_scope
  from tensorflow.python.ops import while_loop
  from tensorflow.python.platform import test
  from tensorflow.python.tpu import tpu_feed
label_desc: Use image node_id 2 for calling the XlaCompileTest obj's underlying member method code with example usage: obj.test_xla_compile_in_function() and returns: xla, unknown