tf.contrib.nccl

 

upper level

 

Modules

 

None

 

Classes

 

None

 

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.