我是Apache TVM框架的新手,我正在为教育和未来使用做一些尝试。我注意到它在 NN 编译和调优方面提供了许多强大的功能。引起我注意的机制之一是张量表达式(TE)。
然而,所有教程主要集中在设置调度程序和生成底层 IR 代码,使用基本示例。我想知道是否有任何方法可以在现有NN 的过程中使用它 - 例如 ResNet,在编译/自动调整过程中作为示例提供,以便利用并行化(例如,通过 OpenCL)或任何优化。
我希望有某种调度绑定机制,允许基于自定义调度的 NN 执行内部运行的操作进行计算,但是我不清楚这种用法。我在这里想念什么吗?