Searched refs:_HostReduceScatter (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | reduce_scatter_cpu_kernel.h | 42 MS_REG_CPU_KERNEL(_HostReduceScatter, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(k…
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_comm_ops.py | 27 … ReduceScatter, _HostReduceScatter, _VirtualDiv, _VirtualAdd, AllSwap,
297 host_all_gather_grad = _HostReduceScatter(ReduceOp.SUM, self.group)
342 @bprop_getters.register(_HostReduceScatter)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 41 … _HostAllGather, _HostReduceScatter, _MirrorMicroStepOperator, _MicroStepAllGather)
|
| D | comm_ops.py | 445 class _HostReduceScatter(PrimitiveWithInfer): class
|