TensorFlow XLA是否已弃用

Question

我刚刚读到(这里(：

不推荐使用此版本的XLA_CPU和XLA_GPU设备。(TF 2.2.0(

XLA现在是否已弃用？它不再使用了？有其他选择吗？还是这一切都是内部自动处理的？这种贬低意味着什么？

Answer 1

不，XLA没有被弃用，事实上还有其他关于它的发行说明：

• XLA
  • XLA现在构建并在窗口上工作。所有预构建软件包均提供XLA
  • XLA可以在CPU和GPU上为具有"编译或抛出异常"语义的tf.function启用

此注释实际上与您提到的折旧有关。不推荐使用的是将字符串XLA_CPU和XLA_GPU与tf.device一起使用。请参阅弃用警告消息：

XLA_GPU and XLA_CPU devices are deprecated and will be
removed in subsequent releases. Instead, use either
@tf.function(experimental_compile=True) for must-compile
semantics, or run with TF_XLA_FLAGS=--tf_xla_auto_jit=2
for auto-clustering best-effort compilation.

这就是另一个发行说明所讨论的内容，在XLA指南中tf.function.的显式编译中提到了这一点