list_dps¶
- torchdata.dataloader2.graph.list_dps(graph: Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], DataPipeGraph]]]], exclude_dps: Optional[Union[IterDataPipe, MapDataPipe, List[Union[IterDataPipe, MapDataPipe]]]] = None) List[Union[IterDataPipe, MapDataPipe]] ¶
Given the graph of DataPipe generated by
traverse_dps
function, return a list of all DataPipe instances without duplication. Ifexclude_dps
is provided, the providedDataPipes
and their predecessors will be ignored.Note
The returned list is in the order of breadth first search of the graph