Python 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 構建的。