| /aosp_15_r20/external/pytorch/test/distributed/checkpoint/ |
| H A D | test_state_dict_utils.py | 7 import torch.distributed._functional_collectives as funcol namespace
|
| /aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/ |
| H A D | test_tp_random_state.py | 3 import torch.distributed._functional_collectives as funcol namespace
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/ |
| H A D | _redistribute.py | 8 import torch.distributed._functional_collectives as funcol namespace
|
| H A D | _collective_utils.py | 9 import torch.distributed._functional_collectives as funcol namespace
|
| H A D | placement_types.py | 8 import torch.distributed._functional_collectives as funcol namespace
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_random_ops.py | 7 import torch.distributed._functional_collectives as funcol namespace
|
| H A D | test_embedding_ops.py | 29 funcol = torch.ops.c10d_functional variable
|
| H A D | test_redistribute.py | 19 funcol = torch.ops.c10d_functional variable
|
| H A D | test_math_ops.py | 33 funcol = torch.ops.c10d_functional variable
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_fake_pg.py | 8 import torch.distributed._functional_collectives as funcol namespace
|
| H A D | test_c10d_functional_native.py | 8 import torch.distributed._functional_collectives as funcol namespace
|
| H A D | test_device_mesh.py | 6 import torch.distributed._functional_collectives as funcol namespace
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/experimental/ |
| H A D | test_local_map.py | 6 import torch.distributed._functional_collectives as funcol namespace
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/debug/ |
| H A D | test_comm_mode.py | 5 import torch.distributed._functional_collectives as funcol namespace
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ |
| H A D | _embedding_ops.py | 9 import torch.distributed._functional_collectives as funcol namespace
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/ |
| H A D | loss.py | 8 import torch.distributed._functional_collectives as funcol namespace
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_mixed_precision.py | 9 import torch.distributed._functional_collectives as funcol namespace
|
| H A D | test_fully_shard_training.py | 59 funcol = torch.ops.c10d_functional variable
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/ |
| H A D | test_2d_composability.py | 102 funcol = torch.ops.c10d_functional variable in TestFullyShard2DTraining
|
| /aosp_15_r20/external/pytorch/torch/distributed/_symmetric_memory/ |
| H A D | __init__.py | 10 import torch.distributed._functional_collectives as funcol namespace
|