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

时间:2017-07-17 14:06:50

标签: tensorflow tensorflow-xla

在训练期间,tensorflow操作如何降低到XLA操作?

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

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

由于

1 个答案:

答案 0 :(得分:0)

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