tf.test.is_built_with_xla

Возвращает, было ли TensorFlow построено с поддержкой XLA.

tf.test.is_built_with_xla()

Этот метод следует использовать только в тестах, написанных с помощью tf.test.TestCase. Типичное использование — пропускать тесты, которые должны выполняться только с XLA.

class MyTest(tf.test.TestCase):

  def test_add_on_xla(self):
    if not tf.test.is_built_with_xla():
      self.skipTest("test is only applicable on XLA")

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

Официальная двоичная версия TensorFlow построена с XLA.