简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
TVM基本调度原语
TVM调度原语官方DOC#!/usr/bin/env python# coding: utf-8# 通常有几种方法来计算相同的结果,但是不同的方法会导致不同的局部性和性能。所以TVM要求用户提供如何执行称为Schedule的计算。# Schedule是一组转换计算的集合,它转换程序中的计算循环。# In[1]:#兼容python2.xfrom __future__ import print_fun
TVM扫描和递归核(Scan and Recurrent Kernel)
扫描和递归核(Scan and Recurrent Kernel)#!/usr/bin/env python# coding: utf-8# # 扫描和递归核(Scan and Recurrent Kernel)# 递归计算是神经网络的一种典型模式。# In[1]:from __future__ import absolute_import, print_functionimport tvmimp
TVM外部张量函数(External Tensor Functions)
#!/usr/bin/env python# coding: utf-8# # 外部张量函数(External Tensor Functions)## 虽然TVM支持透明的代码生成,但有时将手工编写的代码合并到流水线中也很有帮助。例如,我们可能想用cuDNN来做一些卷积内核,并定义其余的阶段。## TVM原生支持这些黑盒函数调用。确切地说,TVM支持所有与DLPack兼容的张量函数。这意味着我们可
暂无文章信息