tvm-rust :(合并)TVM运行时的Rust绑定

时间:2024-02-23 06:52:15
【文件属性】:

文件名称:tvm-rust :(合并)TVM运行时的Rust绑定

文件大小:47KB

文件格式:ZIP

更新时间:2024-02-23 06:52:15

deep-learning compiler rust-library nnvm tvm

已淘汰 RFC已关闭,并且已合并到。 TVM运行时前端支持 此板条箱为运行时前端提供了惯用的Rust API,作为 。 目前,这需要Nightly Rust 。 签出。 此板条箱提供什么? 这是主要的工作流程 使用 , 或等任何主要框架训练您的深度学习模型 使用TVM在受支持的上下文(例如CPU,GPU,OpenCL,Vulkan,VPI,ROCM等)上构建优化的模型工件。 使用Rust部署模型 :red_heart_selector: 示例:从ImageNet1k上的预训练Resnet18部署图像分类 请查看以获取完整的端到端示例。 这是一个Python片段,用于通过MXNet和TVM下载和构建经过预训练的Resnet18 block = get_model ( 'resnet18_v1' , pretrained = True ) sym , params = nnvm . frontend . from_mxnet ( block ) # add the softmax layer for prediction net = nnvm . sym . softm


【文件预览】:
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)
--------errors.rs(1KB)
--------value.rs(13KB)
--------lib.rs(2KB)
----.travis.yml(61B)
----LICENSE(11KB)
----Cargo.toml(591B)
----examples()
--------resnet()
----README.md(8KB)
----tests()
--------callback()
--------basics()
----rustfmt.toml(1KB)
----tvm-sys()
--------src()
--------LICENSE(11KB)
--------Cargo.toml(189B)
--------build.rs(1KB)

网友评论