如何从XLA中的TensorFlow获取*整个*计算图的LLVM IR?

时间:2017-03-11 12:09:41

标签: compilation tensorflow

昨天我posted关于从TensorFlow图获取LLVM IR。我在LOG(INFO) << llvm_ir::DumpModuleToString(*llvm_module)中插入了CpuCompiler::Compile()并获得了LLVM IR。但这似乎只是整个计划的一小部分。

我还应该解释一下大局。请访问this链接,它解释了我想要在XLA编译器中实现的内容。

如何获得整个TensorFlow(XLA)计算图的IR? Dev Summit中的视频表明XLA可能没有整个TensorFlow图,因为一些TensorFlow操作(内核)尚未在XLA中实现。

0 个答案:

没有答案