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 tensor that can be efficiently transferred to other devices.

reduce_sum(...): Returns a tensor with the reduce sum across tensors.