W3cubDocs

/TensorFlow Python

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.

© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/nccl/broadcast