Relay IR的核心数据结构

最新推荐文章于 2025-04-26 11:07:26 发布

胡乱儿起个名

最新推荐文章于 2025-04-26 11:07:26 发布

阅读量859

点赞数 33

分类专栏： TVM 文章标签： TVM 编译器 AI编译器人工智能

本文链接：https://blog.csdn.net/qq_44815135/article/details/147434049

版权

在 Apache TVM 的 Relay IR 中，基础节点（Var、Const、Call、Function 和 Expr）是构建计算图的核心数据结构。以下是对它们的详细解析，包括定义、作用、内部组成及相互关系：

字段/方法	说明
`checked_type_`	表达式的推断类型（如 `TensorType`、`TupleType`）。
`span`	源代码位置信息（用于调试和错误报告）。
`VisitAttrs(visitor)`	递归访问所有属性和子节点（用于序列化、优化等）。
`Mutate()`	生成表达式的副本（用于变换和优化）。