Home
last modified time | relevance | path

Searched refs:DistributedSampler (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/data/samplers/
H A Ddistributed.h26 class DistributedSampler : public Sampler<BatchRequest> {
28 DistributedSampler(
72 class TORCH_API DistributedRandomSampler : public DistributedSampler<> {
105 class TORCH_API DistributedSequentialSampler : public DistributedSampler<> {
/aosp_15_r20/external/pytorch/torch/distributed/elastic/utils/data/
H A Delastic_distributed_sampler.py13 from torch.utils.data.distributed import DistributedSampler
16 class ElasticDistributedSampler(DistributedSampler):
/aosp_15_r20/external/executorch/examples/llm_pte_finetuning/
H A Dtraining_lib.py16 from torch.utils.data import DataLoader, Dataset, DistributedSampler
127 sampler = DistributedSampler(
/aosp_15_r20/external/pytorch/torch/csrc/api/src/data/samplers/
H A Ddistributed.cpp20 : DistributedSampler(size, num_replicas, rank, allow_duplicates), in DistributedRandomSampler()
102 : DistributedSampler(size, num_replicas, rank, allow_duplicates), in DistributedSequentialSampler()
/aosp_15_r20/external/pytorch/torch/utils/data/
H A D__init__.py32 from torch.utils.data.distributed import DistributedSampler
H A Ddistributed.py16 class DistributedSampler(Sampler[_T_co]): class
/aosp_15_r20/external/pytorch/test/
H A Dtest_dataloader.py2209 from torch.utils.data.distributed import DistributedSampler
2213 sampler = DistributedSampler(dataset, 3, 3)
2216 sampler = DistributedSampler(dataset, 3, -1)
2219 from torch.utils.data.distributed import DistributedSampler
2226 s = DistributedSampler(data_set, num_processes, i)
/aosp_15_r20/external/pytorch/docs/source/
H A Ddata.rst442 .. autoclass:: torch.utils.data.distributed.DistributedSampler
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py93 from torch.utils.data.distributed import DistributedSampler
6589 dist_sampler = DistributedSampler(dataset=dataset, drop_last=True)
6622 dist_sampler_added_samples = DistributedSampler(dataset=dataset)
6639 dist_sampler_added_samples_tiny = DistributedSampler(dataset=dataset_tiny)