def gen_function(func_attrs):
    """Generates function body.

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

    Returns
    -------
    str
        Rendered function body.
    """
    return identity_common.gen_function(
        func_attrs=func_attrs,
        backend_spec=ROCMSpec(),
        extra_headers=EXTRA_HEADERS.render(),
    )
Generates function body.

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

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