Searched defs:LoadPlanner (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/torch/distributed/checkpoint/ |
| D | state_dict_loader.py | 35 planner: Optional[LoadPlanner] = None,
58 planner: Optional[LoadPlanner] = None,
192 planner: Optional[LoadPlanner] = None,
|
| D | _checkpointer.py | 40 load_planner: Optional[LoadPlanner] = None,
|
| D | optimizer.py | 218 planner: Optional[LoadPlanner] = None,
|
| D | format_utils.py | 73 def read_data(self, plan: LoadPlan, planner: LoadPlanner) -> Future[None]:
|
| D | storage.py | 256 def read_data(self, plan: LoadPlan, planner: LoadPlanner) -> Future[None]:
|
| D | planner.py | 271 class LoadPlanner: class
|
| D | filesystem.py | 633 def read_data(self, plan: LoadPlan, planner: LoadPlanner) -> Future[None]:
|
| /external/pytorch/test/distributed/checkpoint/ |
| D | test_checkpoint.py | 219 def read_data(self, plan: LoadPlan, planner: LoadPlanner) -> Future[None]:
|