code: def testExampleXLA(self):
      self.skip_if_no_xla()
      arg = tf.constant([1, 2, 3])
      oracle = tf.math.cumsum(arg)
      with tf.device("CPU"):
          ans, runtime = self.evaluate(
              tf.function(cumsum, jit_compile=True)(arg)
          )
      self.assertAllEqual(oracle, ans)
      self.assertEqual(
          implementation_selection._RUNTIME_DEFAULT, runtime
      )
  
file_path: probability/tensorflow_probability/python/internal/implementation_selection_test.py
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.internal import implementation_selection
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the ImplSelectTest obj's underlying member method code with example usage: obj.testExampleXLA() without return types