Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_gpu_kernel.h36 typedef ncclResult_t (*AllGather)(const void *, void *, size_t, ncclDataType_t, cudaStream_t, const… typedef
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/distribution/
Dcollective_wrapper.cc42 ncclResult_t AllGather(const void *input_addr, void *output_addr, size_t count, ncclDataType_t data… in AllGather() function
Dnccl_wrapper.cc57 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/
Dmpi_adapter.cc250 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/
Dcomm_ops.py169 class AllGather(PrimitiveWithInfer): class