Searched defs:prepare_global_plan (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/distributed/checkpoint/ | ||
| D | storage.py | 89 def prepare_global_plan(self, plans: List[SavePlan]) -> List[SavePlan]: member in StorageWriter |
| D | format_utils.py | 136 def prepare_global_plan(self, global_plan: List[LoadPlan]) -> List[LoadPlan]: member in BroadcastingTorchSaveReader |
| D | filesystem.py | 500 def prepare_global_plan(self, plans: List[SavePlan]) -> List[SavePlan]: member in _FileSystemWriter |
| /external/pytorch/test/distributed/checkpoint/ | ||
| D | test_checkpoint.py | 182 def prepare_global_plan(self, plans: List[SavePlan]) -> List[SavePlan]: member in FaultyStorageWriter |