如何为TensorFlow构建MLIR

Question

有没有使用MLIR构建TensorFlow的指南/示例？

我正在尝试了解如何在TensorFlow生态系统中使用TensorFlow和XLA方言。

Answer 1

不幸的是，目前还没有指南。如果你使用Tflite转换器，那么你就是在使用TensorFlow和MLIR，你可以在管道构造中看到它是如何使用的，尽管围绕这一点的指南是专注于用户而不是开发人员。对于XLA，我们正在积极地推出使用MLIR的新的TF XLA桥，但它还没有准备好用于一般的外部使用。使用MLIR的新的XLA HLO发射器也是如此。我不确定TFRT，但我假设它同样会专注于用户，而不是开发人员。

这是最近的文档冲刺中的一个项目，但不幸的是，这项任务没有完成，我们仍然必须解决这个问题。该文档最终将出现在tensorflow.org/mlir上。一旦完成，我会尝试并记住在这里ping。