contrib.nccl.broadcast

tf.contrib.nccl.broadcast

tf.contrib.nccl.broadcast

broadcast(
    src_tensor,
    dst_devices
)

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

Returns a list of tensors on dst_devices, each with value tensor.

The computation is done with a broadcast nccl operation, so if only some of the returned tensors and src_tensor are evaluated then the computation will hang.

Args:

• src_te
  登录查看完整内容