Home
last modified time | relevance | path

Searched defs:c10d_ops (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/_tensor/debug/
H A Dtest_comm_mode.py17 c10d_ops = torch.ops.c10d variable
/aosp_15_r20/external/pytorch/torch/distributed/tensor/debug/
H A D_comm_mode.py29 c10d_ops = torch.ops.c10d variable
/aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/
H A Dtest_2d_composability.py101 c10d_ops = torch.ops.c10d variable in TestFullyShard2DTraining
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_comm.py58 c10d_ops = torch.ops.c10d variable
H A Dtest_fully_shard_training.py58 c10d_ops = torch.ops.c10d variable