tf.contrib.nccl.broadcast

tf.contrib.nccl.broadcast(tensor)

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

Returns a tensor that can be efficiently transferred to other devices.

Args:

  • tensor: The tensor to send; must be assigned to a GPU device.

Returns:

A tensor with the value of src_tensor, which can be used as input to ops on other GPU devices.