【TVM学习一】About TVM

7 篇文章 18 订阅 ¥29.90 ¥99.00
2 篇文章 0 订阅
TVM是一个深度学习编译器,用于CPU、GPU和专用加速器,旨在提升模型在不同硬件上的性能。它能将多种框架的模型编译为最小部署模块,并进行张量运算符的优化。TVM包含计算图优化和张量运算优化两层,提供了高级运算符融合、布局转换和内存管理等功能。
摘要由CSDN通过智能技术生成

    TVM是一个用于CPU, GPU和专用加速器的开放式深度学习编译器堆栈。它旨在缩小以生产力为重点的深度学习框架与面向性能或效率的硬件后端之间的差距。TVM提供以下主要功能:

  •     将Kears, MxNet, Tensorflow, CoreML, DarkNet中的深度学习模型汇编成各种硬件后端的最小可部署模块
  •     基础架构可以在更多的后端自动生成优化张量运算符(tensor operators),并具有更好的性能。

 

    TVM堆栈始于华盛顿大学计算机科学与工程Paul G. Allen学院的SAMPL小组的研究项目。该项目现在由一个涉及多个行业和学术机构的开源社区推动,并采用基于Apache-style的优点进行模型管理。

 

    TVM提供了两个级别的优化,如下图所示 (引自https://tvm.ai/about,侵删):

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值