Searched refs:multiprocessing_context (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/data/ |
| H A D | dataloader.py | 245 multiprocessing_context=None, argument
283 self.multiprocessing_context = multiprocessing_context
418 def multiprocessing_context(self): member in DataLoader
421 @multiprocessing_context.setter
422 def multiprocessing_context(self, multiprocessing_context): argument
423 if multiprocessing_context is not None:
425 if isinstance(multiprocessing_context, str):
427 if multiprocessing_context not in valid_start_methods:
433 multiprocessing_context = torch.multiprocessing.get_context(
434 multiprocessing_context
[all …]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_dataloader.py | 877 multiprocessing_context=mp_ctx,
1367 for multiprocessing_context in supported_multiprocessing_contexts:
1373 multiprocessing_context=multiprocessing_context,
1492 self.dataset, num_workers=0, multiprocessing_context=valid_ctx
1498 self.dataset, num_workers=1, multiprocessing_context="bad"
1504 self.dataset, num_workers=1, multiprocessing_context=object()
1871 multiprocessing_context=ctx,
1884 multiprocessing_context=ctx,
1915 datapipe, multiprocessing_context=ctx, **dl_common_args
1927 datapipe, multiprocessing_context=ctx, **dl_common_args
[all …]
|
| H A D | test_datapipe.py | 2986 multiprocessing_context="spawn",
3000 multiprocessing_context="spawn",
|