Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_gpu_kernel.h37 typedef ncclResult_t (*ReduceScatter)(const void *, void *, size_t, ncclDataType_t, ncclRedOp_t, cu… typedef
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/distribution/
Dcollective_wrapper.cc47 ncclResult_t ReduceScatter(const void *input_addr, void *output_addr, size_t count, ncclDataType_t … in ReduceScatter() function
Dnccl_wrapper.cc65 ncclResult_t NCCLWrapper::ReduceScatter(const void *input_addr, void *output_addr, size_t count, in ReduceScatter() function in mindspore::device::gpu::NCCLWrapper
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/mpi/
Dmpi_adapter.cc163 bool MPIAdapter::ReduceScatter(const float *input, float *output, const std::vector<int> &ranks_gro… in ReduceScatter() function in mindspore::device::cpu::MPIAdapter
/third_party/mindspore/mindspore/ops/operations/
Dcomm_ops.py367 class ReduceScatter(PrimitiveWithInfer): class