【TVM教程】转换
📝
内容提要
我们也可以定义自己的 Pass。例如，我们可以将程序中的relu操作符重写为gelu操作符。首先，我们需要编写一个 Relax IR Mutator 来执行这个替换操作。wp:code@mutator# 访问 relax.Call 表达式，并且只处理当操作（op）是 relax.nn.relu 的情况。/wp:code然后我们可以编写一个 Pass，将这个 Mutator...
🏷️
