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 `input` de `num_devices` usando `reduction` para um único dispositivo.