tvm rust :(合并)TVM运行时的Rust绑定 源码
已淘汰 RFC已关闭,并且已合并到。 TVM运行时前端支持 此板条箱为运行时前端提供了惯用的Rust API,作为 。 目前,这需要Nightly Rust 。 签出。 此板条箱提供什么? 这是主要的工作流程 使用 , 或等任何主要框架训练您的深度学习模型 使用TVM在受支持的上下文(例如CPU,GPU,OpenCL,Vulkan,VPI,ROCM等)上构建优化的模型工件。 使用Rust部署模型 :red_heart_selector: 示例:从ImageNet1k上的预训练Resnet18部署图像分类 请查看以获取完整的端到端示例。 这是一个Python片段,用于通过MXNet和TV
文件列表
tvm-rust-master.zip
(预估有个37文件)
tvm-rust-master
.gitignore
148B
src
ty.rs
6KB
function.rs
16KB
bytearray.rs
2KB
ndarray.rs
12KB
context.rs
8KB
module.rs
3KB
internal_api.rs
794B
暂无评论