| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_ops_nccl.py | 168 def allreduce(tensors, op): function
879 def allreduce(tensors): function
|
| H A D | test_functional_api.py | 571 def allreduce(t, pg): function
581 def allreduce(t, pg): function
|
| H A D | test_c10d_pypg.py | 76 def allreduce(self, tensors, opts): member in LonelyRankProcessGroup
|
| H A D | test_c10d_common.py | 1570 def allreduce(self, tensor_list, opts=None): member in DummyProcessGroup
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| H A D | c_api_coordination_test.cc | 155 TFE_Op* allreduce = AllReduceOp(ctx, in, cluster_size); in TEST() local
380 TFE_Op* allreduce = AllReduceOp(ctx, in, cluster_size); in TEST() local
|
| H A D | c_api_multi_client_test.cc | 99 TFE_Op* allreduce = AllReduceOp(ctx, in, cluster_size); in TEST() local
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | cpp_c10d_extension.cpp | 32 c10::intrusive_ptr<Work> ProcessGroupTest::allreduce( in allreduce() function in c10d::ProcessGroupTest
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | PyProcessGroup.hpp | 95 c10::intrusive_ptr<Work> allreduce( in allreduce() function in c10d::PyProcessGroup
|
| H A D | FakeProcessGroup.hpp | 30 c10::intrusive_ptr<Work> allreduce( in allreduce() function in c10d::FakeProcessGroup
|
| H A D | Backend.hpp | 90 virtual c10::intrusive_ptr<Work> allreduce( in allreduce() function in c10d::Backend
|
| H A D | ProcessGroup.hpp | 154 virtual c10::intrusive_ptr<Work> allreduce( in allreduce() function in c10d::ProcessGroup
|
| H A D | ProcessGroupWrapper.cpp | 406 c10::intrusive_ptr<Work> ProcessGroupWrapper::allreduce( in allreduce() function in c10d::ProcessGroupWrapper
|
| H A D | ProcessGroupGloo.cpp | 1042 void allreduce(std::vector<at::Tensor>& tensors) { in allreduce() function in c10d::__anon2d5ad4bf0911::AsyncAllreduceWork
1199 at::Tensor allreduce(std::vector<at::Tensor>& tensors) { in allreduce() function in c10d::__anon2d5ad4bf0911::AsyncSparseAllreduceWork
1486 c10::intrusive_ptr<Work> ProcessGroupGloo::allreduce( in allreduce() function in c10d::ProcessGroupGloo
|
| H A D | ProcessGroupMPI.cpp | 416 c10::intrusive_ptr<Work> ProcessGroupMPI::allreduce( in allreduce() function in c10d::ProcessGroupMPI
|
| H A D | ProcessGroupUCC.cpp | 997 c10::intrusive_ptr<Work> ProcessGroupUCC::allreduce( in allreduce() function in c10d::ProcessGroupUCC
|
| H A D | ProcessGroupNCCL.cpp | 3436 c10::intrusive_ptr<Work> ProcessGroupNCCL::allreduce( in checkForNCCLErrorsInternal() function in c10d::ProcessGroupNCCL
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | multi_threaded_pg.py | 336 def allreduce(self, tensor_list, opts=AllreduceOptions()): member in ProcessLocalGroup
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_fsdp.py | 224 def allreduce(self, *args, **kwargs): member in DummyProcessGroup
|