Searched refs:all_sum (Results 1 – 4 of 4) sorted by relevance
133 self._Test(partial(_NcclAllReduce, nccl_ops.all_sum), lambda x, y: x + y)140 partial(_NcclAllReduce, nccl_ops.all_sum), lambda x, y: x + y)144 nccl_ops.all_sum([array_ops.identity(np.random.random_sample((3, 4)))])146 nccl_ops.all_sum([])187 all_reduce_tensors = _NcclAllReduce(nccl_ops.all_sum, tensors, devices)
33 def all_sum(tensors): function
43 agg_grads = nccl_ops.all_sum(single_grads)391 summed_grads = nccl_ops.all_sum(scaled_grads)
696 output_tensors = nccl_ops.all_sum(input_tensors)