find_dps

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

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