def METHOD_NAME(x, axis=-1):
    """Compute softmax with MIOpen
    Parameters
    ----------
    x : tvm.te.Tensor
        The input tensor
    axis : int
        The axis to compute softmax over
    Returns
    -------
    ret : tvm.te.Tensor
        The result tensor
    """
    return te.extern(
        x.shape,
        [x],
        lambda ins, outs: tvm.tir.call_packed(
            "tvm.contrib.miopen.softmax.forward", ins[0], outs[0], axis
        ),
        name="y",
    )