Searched defs:LoadPlan (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/checkpoint/ |
| H A D | format_utils.py | 73 def read_data(self, plan: LoadPlan, planner: LoadPlanner) -> Future[None]:
132 def prepare_local_plan(self, plan: LoadPlan) -> LoadPlan:
136 def prepare_global_plan(self, global_plan: List[LoadPlan]) -> List[LoadPlan]:
|
| H A D | storage.py | 224 def prepare_local_plan(self, plan: LoadPlan) -> LoadPlan:
239 def prepare_global_plan(self, plans: List[LoadPlan]) -> List[LoadPlan]:
256 def read_data(self, plan: LoadPlan, planner: LoadPlanner) -> Future[None]:
|
| H A D | planner.py | 103 class LoadPlan: class
367 def create_global_plan(self, global_plan: List[LoadPlan]) -> List[LoadPlan]:
375 def finish_plan(self, central_plan: LoadPlan) -> LoadPlan:
|
| H A D | default_planner.py | 237 def create_global_plan(self, global_plan: List[LoadPlan]) -> List[LoadPlan]:
240 def finish_plan(self, new_plan: LoadPlan) -> LoadPlan:
371 all_plans: List[LoadPlan],
|
| H A D | filesystem.py | 633 def read_data(self, plan: LoadPlan, planner: LoadPlanner) -> Future[None]:
692 def prepare_local_plan(self, plan: LoadPlan) -> LoadPlan:
695 def prepare_global_plan(self, plans: List[LoadPlan]) -> List[LoadPlan]:
|
| /aosp_15_r20/external/pytorch/test/distributed/checkpoint/ |
| H A D | test_checkpoint.py | 211 def prepare_local_plan(self, plan: LoadPlan) -> LoadPlan:
215 def prepare_global_plan(self, plans: List[LoadPlan]) -> List[LoadPlan]:
219 def read_data(self, plan: LoadPlan, planner: LoadPlanner) -> Future[None]:
|