def gen_function_decl(func_attrs, backend_spec):
    """Function declaration generation

    Parameters
    ----------
    func_attrs : Dict[str, Any]
        It describes the operation attributes
    backend_spec : custom class
        It specifies the corresponding backend dtypes of pytorch dtypes for many operations

    Returns
    -------
    str
        Rendered function declaration stmt
    """
    return FUNC_DECL_TEMPLATE.render(
        index_type=backend_spec.index_type,
        prefix=backend_spec.prefix,
        func_name=func_attrs["name"],
    )
Function declaration generation

Parameters
----------
func_attrs : Dict[str, Any]
It describes the operation attributes
backend_spec : custom class
It specifies the corresponding backend dtypes of pytorch dtypes for many operations

Returns
-------
str
Rendered function declaration stmt
python/aitemplate/backend/common/vision_ops/roi_align_common.py