Home
last modified time | relevance | path

Searched defs:SavePlan (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Ddefault_planner.py109 self, all_plans: List[SavePlan]
132 def finish_plan(self, new_plan: SavePlan) -> SavePlan:
410 all_plans: List[SavePlan],
501 def _validate_global_plan(global_plan: List[SavePlan], metadata: Metadata) -> bool:
H A Dstorage.py74 def prepare_local_plan(self, plan: SavePlan) -> SavePlan:
89 def prepare_global_plan(self, plans: List[SavePlan]) -> List[SavePlan]:
107 self, plan: SavePlan, planner: SavePlanner
H A Dplanner.py96 class SavePlan: class
234 self, all_plans: List[SavePlan]
243 def finish_plan(self, new_plan: SavePlan) -> SavePlan:
H A Dfilesystem.py486 def prepare_local_plan(self, plan: SavePlan) -> SavePlan:
500 def prepare_global_plan(self, plans: List[SavePlan]) -> List[SavePlan]:
509 plan: SavePlan,
H A D_dedup_save_plans.py16 all_plans: List[SavePlan],
H A D_dedup_tensors.py34 def dedup_tensors(all_plans: List[SavePlan]) -> List[SavePlan]:
/aosp_15_r20/external/pytorch/test/distributed/checkpoint/
H A Dtest_checkpoint.py178 def prepare_local_plan(self, plan: SavePlan) -> SavePlan:
182 def prepare_global_plan(self, plans: List[SavePlan]) -> List[SavePlan]:
187 self, plan: SavePlan, planner: SavePlanner