code: def testExampleGPU(self):
      self.skip_if_no_xla()
      if not tf.test.is_gpu_available():
          self.skipTest("no GPU")
      arg = tf.constant([1, 2, 3])
      oracle = tf.math.cumsum(arg)
      with tf.device("GPU"):
          ans1, runtime1 = self.evaluate(
              tf.function(cumsum, jit_compile=False)(arg)
          )
          ans2, runtime2 = self.evaluate(
              tf.function(cumsum, jit_compile=True)(arg)
          )
      self.assertAllEqual(oracle, ans1)
      self.assertAllEqual(oracle, ans2)
      self.assertEqual(
          implementation_selection._RUNTIME_DEFAULT, runtime1
      )
      self.assertEqual(
          implementation_selection._RUNTIME_DEFAULT, runtime2
      )
  
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 3 for calling the ImplSelectTest obj's underlying member method code with example usage: obj.testExampleGPU() without return types