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.

登录查看完整内容