def _get_activation_fn(activation):
    """Just a wrapper to get the activation functions.
    """

    if activation == "relu":
        return F.relu
    elif activation == "gelu":
        return F.gelu
Just a wrapper to get the activation functions.
    
clearvoice/clearvoice/models/av_mossformer2_tse/mossformer/utils/one_path_flash_fsmn.py