Searched defs:set_up_storage_writer (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/distributed/checkpoint/ | ||
| D | storage.py | 64 def set_up_storage_writer(self, is_coordinator: bool) -> None: member in StorageWriter |
| D | filesystem.py | 483 def set_up_storage_writer(self, is_coordinator: bool) -> None: member in _FileSystemWriter |
| /external/pytorch/test/distributed/checkpoint/ | ||
| D | test_checkpoint.py | 175 def set_up_storage_writer(self, is_coordinator: bool) -> None: member in FaultyStorageWriter |