def METHOD_NAME(
    orig_var: Var, call_tir_with_grad: Call, output_grad: Var, ctx: BlockBuilder
) -> relax.Expr:
    return ctx.emit_te(
        func,
        output_grad,
        *call_tir_with_grad.args[1],
        **call_tir_with_grad.attrs.te_grad_kwargs,
        primfunc_name_hint=te_grad_name,
    )