Tensorflow - XLA |如何将TF操作降低到XLA进行培训

Question

在训练期间，tensorflow操作如何降低到XLA操作？

tensorflow / compiler / tf2xla / kernels下的文件只根据ComputationBuilder中的ops定义正向传递/推理。

这个问题与“Tensorflow - XLA | Passing tensors to external functions at runtime”有关，因为我需要在训练和推理期间降低到XLA操作系统以通过张量的TF操作。请分享您对其他问题的看法。

由于

Answer 1

我的理解是，Tensorflow在将图传给tf2xla进行降低之前，先计算了反向传播所需的导数。使用tf2xla将所有用于Tensorflow训练和推理的操作降低到XLA;没有其他机制。