HomeSort by relevance Sort by last modified time
    Searched refs:all_sum (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/contrib/nccl/
__init__.py 20 @@all_sum
33 from tensorflow.contrib.nccl.python.ops.nccl_ops import all_sum
  /external/tensorflow/tensorflow/contrib/nccl/python/ops/
nccl_ops_test.py 133 self._Test(partial(_NcclAllReduce, nccl.all_sum), lambda x, y: x + y)
140 partial(_NcclAllReduce, nccl.all_sum), lambda x, y: x + y)
144 nccl.all_sum([array_ops.identity(np.random.random_sample((3, 4)))])
146 nccl.all_sum([])
186 all_reduce_tensors = _NcclAllReduce(nccl.all_sum, tensors, devices)
nccl_ops.py 33 def all_sum(tensors): function
52 """The gradients for `all_sum`.
55 op: The `all_sum` `Operation` that we are differentiating.
56 grad: Gradient with respect to the output of the `all_sum` op.
59 The gradient with respect to the output of `all_sum`.
  /external/tensorflow/tensorflow/contrib/all_reduce/python/
all_reduce.py 697 output_tensors = nccl.all_sum(input_tensors)

Completed in 88 milliseconds