def remove_no_ops(sorted_graph: List[Tensor]) -> List[Tensor]:
    """Remove no-ops from the graph.

    Parameters
    ----------
    sorted_graph : List[Tensor]
        Input graph

    Returns
    -------
    List[Tensor]
        Graph after remove no-ops
    """
    passes = [
        _remove_id_ops,
        _remove_no_op_concats,
        _remove_no_op_dynamic_slices,
        _remove_no_op_splits,
        _remove_no_op_expands,
        _fuse_expand_elementwise,
    ]
    for f_pass in passes:
        sorted_graph = f_pass(sorted_graph)
    return sorted_graph
Remove no-ops from the graph.

    Parameters
    ----------
    sorted_graph : List[Tensor]
        Input graph

    Returns
    -------
    List[Tensor]
        Graph after remove no-ops
    
python/aitemplate/compiler/transform/remove_no_ops.py