docstr: Tests that resource variable usage is allowed.
code: def test_resource_variable_in_function(self):
      """Tests that resource variable usage is allowed."""
      a = variable_scope.get_variable(
          name="variable_a", use_resource=True, initializer=1
      )
  
      @def_function.function
      def func():
          context = self.create_test_xla_compile_context()
          context.Enter()
          o = a.assign(2)
          context.Exit()
          return o
  
      self.assertEqual(self.evaluate(func()), 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 4 for calling the XLACompileContextTest obj's underlying member method code with example usage: obj.test_resource_variable_in_function() and returns: o