Apache TVM 是一个支持 CPU、GPU 和加速芯片的深度学习编译框架。本文以累积乘积为例,介绍如何在 Relay 中注册新的 TVM 算子,包括属性节点、类型关系、算子计算和策略定义等步骤。
完成下面两步后,将自动完成登录并继续当前操作。