def reduce_mean_gen_function(func_attrs) -> str:
    """the registered function for generating reduce_mean kernel and all of
    its auxiliary functions

    Parameters
    ----------
    func_attrs : Dict[str, Any]
        holds attributes of this reduce_mean op

    Returns
    -------
    str
        returns the rendered code for the complete implementation of this reduce mean op
    """
    return reduce_3d.gen_function(
        func_attrs,
        "cutlass::plus",
        reduce_3d.DEFAULT_PROLOGUE_TEMPLATE,
        EPILOGUE_SCALAR_TEMPLATE,
    )
the registered function for generating reduce_mean kernel and all of
    its auxiliary functions

    Parameters
    ----------
    func_attrs : Dict[str, Any]
        holds attributes of this reduce_mean op

    Returns
    -------
    str
        returns the rendered code for the complete implementation of this reduce mean op
    
python/aitemplate/backend/cuda/reduce/reduce_mean.py