code: def test_trace_only_bypasses_xla(self):
      self.skip_if_no_xla()
      d = jit_public_methods.JitPublicMethods(XLADetector())
      d_trace_only = jit_public_methods.JitPublicMethods(
          XLADetector(), trace_only=True
      )
      self.assertTrue(self.evaluate(d.in_xla_context()))
      self.assertFalse(self.evaluate(d_trace_only.in_xla_context()))
  
imports: import types
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import invert
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import transformed_distribution
  from tensorflow_probability.python.distributions import uniform
  from tensorflow_probability.python.experimental.util import jit_public_methods
  from tensorflow_probability.python.internal import test_util
  from tensorflow.python.ops import control_flow_util
label_desc: Use image node_id 6 for calling the TracedPublicMethodsTest obj's underlying member method code with example usage: obj.test_trace_only_bypasses_xla() without return types