Searched defs:StateDictOptions (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/checkpoint/ |
| H A D | state_dict.py | 105 class StateDictOptions: class
270 options: Optional[StateDictOptions] = None,
967 options: Optional[StateDictOptions] = None,
1005 options: Optional[StateDictOptions] = None,
1050 options: Optional[StateDictOptions] = None,
1171 options: Optional[StateDictOptions] = None,
1211 options: Optional[StateDictOptions] = None,
1251 options: Optional[StateDictOptions] = None,
1311 options: Optional[StateDictOptions] = None,
1367 options: Optional[StateDictOptions] = None,
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | common_state_dict.py | 37 options: StateDictOptions = StateDictOptions(),
|