tf.test.is_cpu_target_available

Indicates whether TensorFlow was built with support for a given CPU target.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.test.is_cpu_target_available

tf.test.is_cpu_target_available(
    target
)

Args
target	The name of the CPU target whose support to check for.

Returns
A boolean indicating whether TensorFlow was built with support for the given CPU target.

This method should only be used in tests written with tf.test.TestCase. A typical usage is to skip tests that should only run with a given target.

class MyTest(tf.test.TestCase):

  def test_add_on_aarch64(self):
    if not tf.test.is_cpu_target_available('aarch64'):
      self.skipTest("test is only applicable on AArch64")

... @tf.function(jit_compile=True) ... def add(x, y): ... return tf.math.add(x, y) ... ... self.assertEqual(add(tf.ones(()), tf.ones(())), 2.0)