docstr: Tests that xla.compile works with variable in tf.function.
code: def test_xla_compile_write_variable_in_function(self):
      """Tests that xla.compile works with variable in tf.function."""
      a = variable_scope.get_variable(
          name="variable_a", use_resource=True, initializer=1
      )
  
      @def_function.function
      def func_wrapper():
          def compute():
              a.assign_add(1)
              a.assign_sub(2)
              return a.read_value()
  
          return xla.compile(compute)
  
      self.evaluate(a.initializer)
      self.assertEqual(self.evaluate(func_wrapper())[0], 0)
  
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 3 for calling the XlaCompileTest obj's underlying member method code with example usage: obj.test_xla_compile_write_variable_in_function() and returns: xla, a