
TVM
丶Shining
深度学习爱好者,深度学习部署优化。
Github:https://github.com/xxradon;
Mail:wangdong_cn@qq.com
展开
-
TVM系列---1.开始使用Tensor Expression
Author: Tianqi Chen https://docs.tvm.ai/tutorials/tensor_expr_get_started.html Tensor Expression入门 这是TVM中Tensor表达语言的入门教程。TVM使用特定于域的张量表达式来进行有效的内核构造。 在本教程中,我们将演示使用张量表达式语言的基本工作流程。 from __future__ import ...翻译 2019-08-02 17:52:05 · 2896 阅读 · 0 评论 -
TVM系列---2.Compile ONNX Models
本文是使用Relay部署ONNX模型的入门教程。 首先,我们必须安装ONNX软件包。 一个快速的解决方案是安装protobuf编译器,和 pip install onnx --user 或者请参考官方网站。 https://github.com/onnx/onnx import onnx import numpy as np import tvm import tvm.relay as rela...翻译 2019-08-02 18:01:56 · 1145 阅读 · 0 评论