Searched defs:dw_builder (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/pipelining/ | ||
| H A D | test_schedule_multiproc.py | 724 def dw_builder(self): member in ScheduleTest.test_schedule_with_weight_update_mlp_e2e.CustomState |
| H A D | test_stage.py | 276 def dw_builder(self): member in StageTest.test_custom_dw_with_fb_schedule.CustomState |