docstr: Tests that XLACompileContext is recognized as an XLA context.
code: def test_op_is_in_context(self):
      """Tests that XLACompileContext is recognized as an XLA context."""
      op1 = constant_op.constant(1)
      context = self.create_test_xla_compile_context()
      context.Enter()
      op2 = constant_op.constant(2)
      context.Exit()
      self.assertFalse(control_flow_util.IsInXLAContext(op1.op))
      self.assertTrue(control_flow_util.IsInXLAContext(op2.op))
  
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 11 for calling the XLACompileContextTest obj's underlying member method code with example usage: obj.test_op_is_in_context() without return types