def METHOD_NAME(model, m_list):
    if isinstance(model, tuple(m_list)):
        return [model]
    res = []
    for mo in model.modules():
        if isinstance(mo, tuple(m_list)):
            res.append(mo)
    return res