摘要:
使用Tensor Expression张量表达式处理算子 这是TVM中Tensor表达语言的入门教程。TVM使用特定于域的张量表达式来进行有效的内核构造。 本文将演示使用张量表达式语言的基本工作流程。 from __future__ import absolute_import, print_fun 阅读全文
摘要:
TVM优化c++部署实践 使用TVM导入神经网络模型: 模型支持pytorch , tensorflow , onnx, caffe 等。平时pytorch用的多,这里给一种pytorch的导入方式。 github代码仓:https://github.com/leoluopy/autotvm_tut 阅读全文