OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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