Searched defs:backend_str (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/ | ||
| H A D | test_pp_composability.py | 55 def backend_str(cls) -> str: member in ComposabilityTest |
| /aosp_15_r20/external/pytorch/test/distributed/pipelining/ | ||
| H A D | test_schedule_multiproc.py | 48 def backend_str(cls) -> str: member in ScheduleTest |
| H A D | test_stage.py | 64 def backend_str(cls) -> str: member in StageTest |
| /aosp_15_r20/external/pytorch/test/distributed/ | ||
| H A D | test_c10d_ops_nccl.py | 48 def backend_str(cls) -> str: member in ProcessGroupNCCLOpTest |
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ | ||
| H A D | common_distributed.py | 1335 def backend_str(cls) -> str: member in MultiProcContinousTest |