def gen_function(func_attrs, element_func=None, element_func_def=None):
    """Generates function body.

    Parameters
    ----------
    func_attrs : Dict[str, Any]
        Stores the operation attributes.

    Returns
    -------
    str
        Rendered function body.
    """
    return concatenate_common.gen_function(
        func_attrs=func_attrs,
        backend_spec=ROCMSpec(),
        element_func=element_func,
        element_func_def=element_func_def,
    )
Generates function body.

    Parameters
    ----------
    func_attrs : Dict[str, Any]
        Stores the operation attributes.

    Returns
    -------
    str
        Rendered function body.
    
python/aitemplate/backend/rocm/tensor/concatenate.py