MLIR 整合了在 TensorFlow 中构建高性能机器学习模型所需的基础架构。

MLIR 项目定义了一种通用中间表示法 (IR),该表示法整合了在 TensorFlow 和类似机器学习框架中执行高性能机器学习模型所需的基础架构。该项目将包括 HPC 技术的应用以及强化学习等搜索算法的集成。MLIR 旨在降低新硬件的开发成本,并针对现有 TensorFlow 用户提高了易用性。
// Syntactically similar to LLVM:
func @testFunction(%arg0: i32) {
  %x = call @thingToCall(%arg0) : (i32) -> i32
  br ^bb1
^bb1:
  %y = addi %x, %x : i32
  return %y : i32
}