Package org.tensorflow.op.distribute

Class Summary

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