Searched refs:_HostAllGather (Results 1 – 4 of 4) sorted by relevance
42 MS_REG_CPU_KERNEL(_HostAllGather, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumb…
25 from ..operations.comm_ops import (AllGather, _MiniStepAllGather, _HostAllGather, AllReduce, Neighb…294 @bprop_getters.register(_HostAllGather)345 host_reduce_scatter_grad = _HostAllGather(self.group)
41 … _HostAllGather, _HostReduceScatter, _MirrorMicroStepOperator, _MicroStepAllGather)
314 class _HostAllGather(PrimitiveWithInfer): class