TVM优化原理学习

没有看论文,看了b站陈天奇的视频,还有一些博客的分析,学习了优化原理。后续需要深入理解再看论文。   TVM对于神经网络的优化主要有两部分,计算图优化和算子优化,下面分开说明。 计算图优化 例如算子融合,常量传播等。这一步是在IR层面完成,现在TVM是使用relay IR来描述计算图。优化的过程蕴含在relay.build中,参考https://zhuanlan.zhihu.com/p/11396
相关文章
相关标签/搜索