Searched defs:IterDataPipe (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/data/datapipes/iter/ |
| H A D | combining.py | 59 def __init__(self, *datapipes: IterDataPipe):
110 datapipe: IterDataPipe,
159 datapipe: IterDataPipe,
336 def __init__(self, main_datapipe: IterDataPipe, instance_id: int):
425 datapipe: IterDataPipe,
690 def __init__(self, *datapipes: IterDataPipe):
|
| H A D | callable.py | 71 datapipe: IterDataPipe,
225 datapipe: IterDataPipe,
|
| H A D | grouping.py | 76 datapipe: IterDataPipe,
133 def __init__(self, datapipe: IterDataPipe, unbatch_level: int = 1):
|
| H A D | filelister.py | 39 root: Union[str, Sequence[str], IterDataPipe] = ".", argument
|
| H A D | sharding.py | 43 def __init__(self, source_datapipe: IterDataPipe, sharding_group_filter=None):
|
| H A D | combinatorics.py | 35 datapipe: IterDataPipe,
|
| /aosp_15_r20/external/pytorch/torch/utils/data/datapipes/utils/ |
| H A D | snapshot.py | 12 datapipe: IterDataPipe, n_iterations: int, rng=None
|
| /aosp_15_r20/external/pytorch/torch/utils/data/datapipes/ |
| H A D | _decorator.py | 79 def __init__(self, arg: Union[Type[IterDataPipe], Callable[[], bool]]) -> None:
|
| H A D | datapipe.py | 54 class IterDataPipe(IterableDataset[_T_co], metaclass=_IterDataPipeMeta): class
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_datapipe.py | 2670 def __init__(self, dp: IterDataPipe):
3566 def _check_single_iterator_invalidation_logic(self, source_dp: IterDataPipe):
|