Searched defs:AllGather (Results 1 – 5 of 5) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/ |
D | nccl_gpu_kernel.h | 36 typedef ncclResult_t (*AllGather)(const void *, void *, size_t, ncclDataType_t, cudaStream_t, const… typedef
|
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/distribution/ |
D | collective_wrapper.cc | 42 ncclResult_t AllGather(const void *input_addr, void *output_addr, size_t count, ncclDataType_t data… in AllGather() function
|
D | nccl_wrapper.cc | 57 ncclResult_t NCCLWrapper::AllGather(const void *input_addr, void *output_addr, size_t count, ncclDa… in AllGather() function in mindspore::device::gpu::NCCLWrapper
|
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/mpi/ |
D | mpi_adapter.cc | 250 bool MPIAdapter::AllGather(const float *input, float *output, const std::vector<int> &ranks_group, … in AllGather() function in mindspore::device::cpu::MPIAdapter
|
/third_party/mindspore/mindspore/ops/operations/ |
D | comm_ops.py | 169 class AllGather(PrimitiveWithInfer): class
|