昨天我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中实现。