TVM Compiler中文教程:TVM.Relay使用外部库

文章目录 TVM.Relay使用外部库 建立一个简单的网络 使用cuda后端构建和运行 卷积层使用cuDNN实现 验证结果 结论 TVM.Relay使用外部库 这篇教程介绍怎么在Relay中使用cuDNN、cuBlas这样的外部库。python Relay在内部使用TVM生成特定目标的代码。例如,使用cuda做为后端,TVM为用户提供的网络生成全部层的cuda kernel代码。可是有时,将设备厂
相关文章
相关标签/搜索