Module: contrib.nccl

Module: tf.contrib.nccl

Module tf.contrib.nccl

Defined in tensorflow/contrib/nccl/__init__.py.

Functions for using NVIDIA nccl collective ops.

Functions

all_max(...): Returns a list of tensors with the all-reduce max across tensors.

all_min(...): Returns a list of tensors with the all-reduce min across tensors.

all_prod(...): Returns a list of tensors with the all-reduce product across tensors.

all_sum(...): Returns a list of tensors with the all-reduce sum across tensors.

broadcast(...): Returns a list of tensors on dst_devices, each with value tensor.