org.tensorflow.op.distribute

Stay organized with collections Save and categorize content based on your preferences.

Classes

NcclAllReduce<T extends TNumber> Outputs a tensor containing the reduction across all input tensors. 
NcclBroadcast<T extends TNumber> Sends `input` to all devices that are connected to the output. 
NcclReduce<T extends TNumber> Reduces `input` from `num_devices` using `reduction` to a single device.