CustomState (definition) in projects: aosp_15_r20

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

/aosp_15_r20/external/pytorch/test/distributed/pipelining/
H A D	test_schedule_multiproc.py	`717 class CustomState: class`
H A D	test_stage.py	`272 class CustomState: class`
/aosp_15_r20/external/pytorch/test/nn/
H A D	test_load_state_dict.py	`225 class CustomState(nn.Module): class`