Searched defs:rootTensor (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/c10d/ |
| H A D | ProcessGroupNCCLTest.cpp | 270 c10::intrusive_ptr<c10d::Work> run(int rootRank, int rootTensor) { in run()
289 c10::intrusive_ptr<c10d::Work> run(int rootRank, int rootTensor) { in run()
537 for (const auto rootTensor : c10::irange(numDevices)) { in testBroadcast() local
564 for (const auto rootTensor : c10::irange(numDevices)) { in testReduce() local
698 for (const auto rootTensor : c10::irange(numDevices)) { in testSplittingCommunicator() local
|
| H A D | ProcessGroupGlooAsyncTest.cpp | 158 c10::intrusive_ptr<c10d::Work> run(int rootRank, int rootTensor) { in run()
230 for (const auto rootTensor : c10::irange(numTensors)) { in runAsyncBroadcastTest() local
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_ops_nccl.py | 103 def broadcast(xs, rootRank, rootTensor): argument
323 def reduce(xs, rootRank, rootTensor, op=None): argument
|
| H A D | test_c10d_ucc.py | 177 def broadcast(xs, rootRank, rootTensor): argument
|
| H A D | test_c10d_gloo.py | 319 def broadcast(xs, rootRank, rootTensor): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | Types.hpp | 116 int64_t rootTensor = 0; member
132 int64_t rootTensor = 0; member
|
| H A D | ProcessGroupGloo.cpp | 875 int rootTensor, in AsyncBroadcastWork()
893 const int rootTensor; member in c10d::__anon2d5ad4bf0711::AsyncBroadcastWork
924 int rootTensor, in AsyncBroadcastCUDAWork()
1625 int rootTensor, in AsyncReduceWork()
1645 const int rootTensor; member in c10d::__anon2d5ad4bf0e11::AsyncReduceWork
1684 int rootTensor, in AsyncReduceCUDAWork()
|