find_dps

torchdata.dataloader2.graph.find_dps(graph: Dict[int, Tuple[IterDataPipe | MapDataPipe, Dict[int, Tuple[IterDataPipe | MapDataPipe, DataPipeGraph]]]], dp_type: Type[IterDataPipe | MapDataPipe]) → List[IterDataPipe | MapDataPipe]

Given the graph of DataPipe generated by traverse_dps function, return DataPipe instances with the provided DataPipe type.