weixin_47325904的博客

原创 TVM.Relay.Var API功能解析

局部变量创建完成后，是可以合入到完整模型的IR图中的。比如想在模型的IR图中改变模型，就可以先创建局部变量，再插入到模型的指定位置，这样新的IR图就生成了。在TVM官方文档的Python API中没有找到TVM.Relay.Var API的介绍，写份文档记录下该函数的功能。在TVM的Realy图中，包含全局变量(IR图中以@标识)和局部变量(IR图中以%标识)。其中2、3、4、5四种写法的输出是一样的，dtype不设为空的话，会默认设为float32。