Searched defs:ScheduleClass (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/pipelining/ |
| H A D | test_schedule_multiproc.py | 65 def test_forward_only(self, ScheduleClass): argument
115 def test_multi_iter(self, ScheduleClass): argument
155 def test_kwargs_with_tracer(self, ScheduleClass): argument
205 def test_grad_with_tracer(self, ScheduleClass, ModelClass): argument
280 def test_grad_with_manual(self, ScheduleClass): argument
357 def test_grad_with_manual_interleaved(self, ScheduleClass, use_new_runtime): argument
500 def test_schedule_with_native_zero_bubble(self, ScheduleClass): argument
587 def test_non_symmetric_stage_ids(self, ScheduleClass): argument
677 def test_schedule_with_weight_update_mlp_e2e(self, ScheduleClass): argument
|
| H A D | test_schedule.py | 126 def test_pipeline_order(self, ScheduleClass): argument
161 def test_pipeline_order_flex_and_zero_bubble(self, ScheduleClass): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/ |
| H A D | test_pp_composability.py | 94 def test_manual_with_data_parallel(self, dp_type, ScheduleClass, use_new_runtime): argument
|