def METHOD_NAME(matched: relay.Call) -> bool:
    """Check if log_softmax is supported by cuDNN."""
    if matched.args[0].checked_type.dtype not in ["float64", "float32", "float16"]:
        return False
    if len(matched.args[0].checked_type.shape) != 2:
        return False
    if matched.attrs["axis"] not in (1, -1):
        return False
    return True