Searched defs:reduce_sum (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/nccl/
H A D__init__.py21 @@reduce_sum
35 from tensorflow.contrib.nccl.python.ops.nccl_ops import reduce_sum namespace
/external/tensorflow/tensorflow/contrib/nccl/python/ops/
H A Dnccl_ops.py130 def reduce_sum(tensors): function
151 """The gradients for input `Operation` of `reduce_sum`.
155 grad: Gradient with respect to the output of the `reduce_sum` op.
158 The gradient with respect to the input of `reduce_sum` op.
202 # Grab inputs of accumulated_grad and replace accumulation with reduce_sum.
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
H A Dops.py973 reduce_sum = define_reduce_op('reduce_sum', math_ops.reduce_sum) variable
/external/tensorflow/tensorflow/contrib/labeled_tensor/
H A D__init__.py131 reduce_sum = _ops.reduce_sum variable
/external/tensorflow/tensorflow/python/ops/
H A Dmath_ops.py113 @@reduce_sum
1318 @tf_export("reduce_sum")
1321 def reduce_sum(input_tensor, function
1341 tf.reduce_sum(x) # 6
1342 tf.reduce_sum(x, 0) # [2, 2, 2]
1343 tf.reduce_sum(x, 1) # [3, 3]
1344 tf.reduce_sum(x, 1, keepdims=True) # [[3], [3]]
1345 tf.reduce_sum(x, [0, 1]) # 6
1437 reduce_sum(
1844 reduce_sum(
[all...]

Completed in 175 milliseconds