tf.contrib.nccl.reduce_sum

tf.contrib.nccl.reduce_sum(tensors)

Defined in tensorflow/contrib/nccl/python/ops/nccl_ops.py.

Returns a tensor with the reduce sum across tensors.

The computation is done with a reduce operation, so only one tensor is returned.

Args:

  • tensors: The input tensors across which to sum; must be assigned to GPU devices.

Returns:

A tensor containing the sum of the input tensors.

Raises:

  • LookupError: If context is not currently using a GPU device.