Searched defs:SavePlanner (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/checkpoint/ |
| H A D | state_dict_saver.py | 40 planner: Optional[SavePlanner] = None,
64 planner: Optional[SavePlanner] = None,
167 planner: Optional[SavePlanner] = None,
269 planner: Optional[SavePlanner] = None,
|
| H A D | _checkpointer.py | 41 save_planner: Optional[SavePlanner] = None,
|
| H A D | filesystem.py | 277 planner: SavePlanner,
510 planner: SavePlanner,
|
| H A D | storage.py | 107 self, plan: SavePlan, planner: SavePlanner
|
| H A D | planner.py | 109 class SavePlanner(abc.ABC): class
|
| /aosp_15_r20/external/pytorch/test/distributed/checkpoint/ |
| H A D | test_checkpoint.py | 187 self, plan: SavePlan, planner: SavePlanner
|