Home
last modified time | relevance | path

Searched defs:prepare_global_plan (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/distributed/checkpoint/
Dstorage.py89 def prepare_global_plan(self, plans: List[SavePlan]) -> List[SavePlan]: member in StorageWriter
239 def prepare_global_plan(self, plans: List[LoadPlan]) -> List[LoadPlan]: member in StorageReader
Dformat_utils.py136 def prepare_global_plan(self, global_plan: List[LoadPlan]) -> List[LoadPlan]: member in BroadcastingTorchSaveReader
Dfilesystem.py500 def prepare_global_plan(self, plans: List[SavePlan]) -> List[SavePlan]: member in _FileSystemWriter
695 def prepare_global_plan(self, plans: List[LoadPlan]) -> List[LoadPlan]: member in FileSystemReader
/external/pytorch/test/distributed/checkpoint/
Dtest_checkpoint.py182 def prepare_global_plan(self, plans: List[SavePlan]) -> List[SavePlan]: member in FaultyStorageWriter
215 def prepare_global_plan(self, plans: List[LoadPlan]) -> List[LoadPlan]: member in FaultyStorageReader