org.tensorflow.op.distribute

Aulas

NcclAllReduce <T estende TNumber > Produz um tensor contendo a redução em todos os tensores de entrada.
NcclBroadcast <T estende TNumber > Envia `entrada` para todos os dispositivos que estão conectados à saída.
NcclReduce <T estende TNumber > Reduz a `entrada` de` num_devices` usando `redução` para um único dispositivo.