- scheduler tvm schedule 详细举例,这里将 https://zhuanlan.zhihu.com/p/94846767 这篇文章的例子用TVM 0.8.0.dev 重写。
- dataflow_controlflow 数据流和控制流的区别示例。
- relay TVM 中一些 Relay 相关的示例,比如如何自定义 Pass,如何在 Jetson Nano 中运行DarkNet的YOLO模型等。
tvm_pytorch_resnet18_inference.py
使用 TVM 在 X86 CPU 上运行 Pytorch 的 ResNet18 模型。pytorch_resnet18_export_onnx.py
Pytorch 导出 ResNet18 的 ONNX 模型示例。tvm_onnx_resnet18_inference.py
TVM 加载 ResNet18 的 ONNX 模型进行推理。
forked from BBuf/tvm_mlir_learn
-
Notifications
You must be signed in to change notification settings - Fork 0
wanghongsheng01/tvm_learn
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
tvm learn
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 100.0%