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