def jagged_to_dense_gen_function_decl(func_attrs) -> str:
    """Generate jagged_to_dense function declaration."""

    matrices = func_attrs["inputs"][1]
    func_name = func_attrs["name"]
    backend_spec = CUDASpec()

    return FUNC_DECL_TEMPLATE.render(
        prefix=backend_spec.prefix,
        index_type=backend_spec.index_type,
        func_name=func_name,
        offsets=gen_offsets_str(
            matrices._attrs["shape"][0],
            has_type=True,
            const_ref=True,
            name="offsets",
        ),
    )
Generate jagged_to_dense function declaration.
python/aitemplate/backend/cuda/gemm_special/batched_dense_vec_jagged_2d_mul.py