在训练期间,tensorflow操作如何降低到XLA操作?
tensorflow / compiler / tf2xla / kernels下的文件只根据ComputationBuilder中的ops定义正向传递/推理。
这个问题与“Tensorflow - XLA | Passing tensors to external functions at runtime”有关,因为我需要在训练和推理期间降低到XLA操作系统以通过张量的TF操作。请分享您对其他问题的看法。
由于
答案 0 :(得分:0)
我的理解是,Tensorflow在将图传给tf2xla进行降低之前,先计算了反向传播所需的导数。使用tf2xla将所有用于Tensorflow训练和推理的操作降低到XLA;没有其他机制。