def squeeze_gen_function_decl(func_attrs):
    """Generate the function declaration for squeeze/unsqueeze.

    Parameters
    ----------
    func_attrs : Dict[str, Any]
        The _attrs dict from the original op.
    """
    func_name = func_attrs["name"]
    input_ndim = len(func_attrs["inputs"][0]._attrs["shape"])
    output_ndim = len(func_attrs["outputs"][0]._attrs["shape"])

    return FUNC_DECL_TEMPLATE.render(
        func_name=func_name, input_ndim=input_ndim, output_ndim=output_ndim
    )
Generate the function declaration for squeeze/unsqueeze.

Parameters
----------
func_attrs : Dict[str, Any]
The _attrs dict from the original op.
python/aitemplate/backend/rocm/view_ops/view_ops.py