Home
last modified time | relevance | path

Searched refs:all_sum (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dnccl_ops_test.py133 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)
Dnccl_ops.py33 def all_sum(tensors): function
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_utils.py43 agg_grads = nccl_ops.all_sum(single_grads)
391 summed_grads = nccl_ops.all_sum(scaled_grads)
Dall_reduce.py696 output_tensors = nccl_ops.all_sum(input_tensors)