code: def testExampleCPU(self):
      arg = tf.constant([1, 2, 3])
      oracle = tf.math.cumsum(arg)
      with tf.device("CPU"):
          ans, runtime = cumsum(arg)
          # TODO(b/153684592): Drop +0, move `self.evaluate` to the line above.
          ans, runtime = self.evaluate([ans + 0, runtime + 0])
      self.assertAllEqual(oracle, ans)
      self.assertEqual(implementation_selection._RUNTIME_CPU, 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 1 for calling the ImplSelectTest obj's underlying member method code with example usage: obj.testExampleCPU() without return types