Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_gpu_kernel.h34 typedef ncclResult_t (*AllReduce)(const void *, void *, size_t, ncclDataType_t, ncclRedOp_t, cudaSt… typedef
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dcollective_ops_impl.cc214 bool CollectiveOpsImpl::AllReduce(const void *sendbuff, void *recvbuff, size_t count) { in AllReduce() function in mindspore::fl::server::CollectiveOpsImpl
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/distribution/
Dcollective_wrapper.cc37 ncclResult_t AllReduce(const void *input_addr, void *output_addr, size_t count, ncclDataType_t data… in AllReduce() function
Dnccl_wrapper.cc49 ncclResult_t NCCLWrapper::AllReduce(const void *input_addr, void *output_addr, size_t count, ncclDa… in AllReduce() function in mindspore::device::gpu::NCCLWrapper
/third_party/mindspore/mindspore/ops/operations/
Dcomm_ops.py92 class AllReduce(PrimitiveWithInfer): class