2021年11月19日
如何将算子添加到Relay
摘要： 如何将算子添加到Relay 本文将介绍在Relay中注册新TVM算子所需的步骤。将PR添加累积产品运算示例。PR本身建立在另一个PR的基础上，该PR添加了一个累积和运算。 注册新算子需要几个步骤： 添加一个属性节点，声明编译时已知的固定参数 为算子编写一个类型关系，集成到Relay的类型系统中。 使 阅读全文
posted @ 2021-11-19 06:27 吴建明wujianming 阅读(148) 评论(0) 推荐(0) 编辑