tf.contrib.nccl.reduce_sum

reduce_sum(
    tensors,
    dst_device
)

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.
• dst_device: The device of the returned tensor.

Returns:

A tensor containing the sum of the input tensors, with the device of the tensor being dst_device.