org.tensorflow.op.distribute

Classi

NcclAllReduce <T estende TNumber > Restituisce un tensore contenente la riduzione su tutti i tensori di input.
NcclBroadcast <T estende TNumber > Invia "input" a tutti i dispositivi collegati all'output.
NcclReduce <T estende TNumero > Riduce "input" da "num_devices" utilizzando "reduction" a un singolo dispositivo.