摘要: 使用Tensor Expression张量表达式处理算子 这是TVM中Tensor表达语言的入门教程。TVM使用特定于域的张量表达式来进行有效的内核构造。 本文将演示使用张量表达式语言的基本工作流程。 from __future__ import absolute_import, print_fun 阅读全文
posted @ 2021-11-10 16:58 吴建明wujianming 阅读(142) 评论(0) 推荐(0) 编辑
摘要: TVM优化c++部署实践 使用TVM导入神经网络模型: 模型支持pytorch , tensorflow , onnx, caffe 等。平时pytorch用的多,这里给一种pytorch的导入方式。 github代码仓:https://github.com/leoluopy/autotvm_tut 阅读全文
posted @ 2021-11-10 06:09 吴建明wujianming 阅读(1121) 评论(0) 推荐(0) 编辑