语言参考#

本文档提供了 TVM 堆栈中的嵌入式语言和 IR 的参考。

Relay 简介#

Relay 是函数式的、可微分的编程语言,旨在成为机器学习系统表达式的中间表示。Relay 支持代数数据类型、闭包、控制流和递归,使其能够直接表示比基于计算图的 IR 更复杂的模型。Relay 还包括使用 类型关系 的依赖类型,以处理对参数形状有复杂要求的算子的形状分析。

Relay 在设计上是可扩展的,使机器学习研究人员和从业人员很容易开发新的大规模程序变换和优化。

下面的页面分别描述了 Relay 中的语法、类型系统、代数数据类型和算子。

混合脚本#

下面的页面描述了 TVM 混合脚本前端,它使用软件仿真来支持 TVM 中不正式支持的一些结构。