首页 > TAG信息列表 > AutoTVM

TVM:使用调度模板和AutoTVM优化算子

本节学习如何使用TVM 张量表达式(TE)语言来编写调度模板,这些模板可以被autoTVM搜索到,以找到最佳调度。这个过程称为auto-Tuning,它有助于优化张量计算的自动化过程。 本节建立在如何使用TE编写矩阵乘法的基础上 auto-tuning的步骤如下: 第一搜索空间 第二运行一个搜索算法来探索这