Searched refs:all_sum (Results 1 – 4 of 4) sorted by relevance
134 self._Test(partial(_NcclAllReduce, nccl_ops.all_sum), lambda x, y: x + y)141 partial(_NcclAllReduce, nccl_ops.all_sum), lambda x, y: x + y)145 nccl_ops.all_sum([array_ops.identity(np.random.random_sample((3, 4)))])147 nccl_ops.all_sum([])188 all_reduce_tensors = _NcclAllReduce(nccl_ops.all_sum, tensors, devices)
33 def all_sum(tensors): function
46 agg_grads = nccl_ops.all_sum(single_grads)
702 output_tensors = nccl_ops.all_sum(input_tensors)