StateDictSettings (definition) in projects: aosp_15_r20

Searched defs:StateDictSettings (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D	api.py	`407 class StateDictSettings: class`
H A D	_optim_utils.py	`2077 state_dict_settings: StateDictSettings,`
/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A D	test_fsdp_optim_state.py	`1746 state_dict_settings: StateDictSettings,`