Searched defs:prepare_local_plan (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/distributed/checkpoint/ | ||
| D | storage.py | 74 def prepare_local_plan(self, plan: SavePlan) -> SavePlan: member in StorageWriter |
| D | format_utils.py | 132 def prepare_local_plan(self, plan: LoadPlan) -> LoadPlan: member in BroadcastingTorchSaveReader |
| D | filesystem.py | 486 def prepare_local_plan(self, plan: SavePlan) -> SavePlan: member in _FileSystemWriter |
| /external/pytorch/test/distributed/checkpoint/ | ||
| D | test_checkpoint.py | 178 def prepare_local_plan(self, plan: SavePlan) -> SavePlan: member in FaultyStorageWriter |