【TVM教程】 自定义relay算子
本文为tvm 教程的翻译版。这部分介绍了如何在tvm中添加新的relay算子,具体的是以一个累乘(cumprod)算子为例进行介绍。 新增relay算子基本是下面几个步骤: 定义新增算子的属性节点(Attribute Node),声明在编译时已知的固定参数 为新增算子编写类型关系 ...
本文为tvm 教程的翻译版。这部分介绍了如何在tvm中添加新的relay算子,具体的是以一个累乘(cumprod)算子为例进行介绍。 新增relay算子基本是下面几个步骤: 定义新增算子的属性节点(Attribute Node),声明在编译时已知的固定参数 为新增算子编写类型关系 ...
为实现多种语言支持,需要满足以下几点: 部署:编译结果可以从python/javascript/c++调用。 Debug: 在python中定义一个函数,在编译函数中调用。 链接:编写 ...