org.tensorflow.op.distribute

Des classes

NcclAllReduce <T étend TNumber > Génère un tenseur contenant la réduction sur tous les tenseurs d'entrée.
NcclBroadcast <T étend TNumber > Envoie « entrée » à tous les appareils connectés à la sortie.
NcclReduce <T étend TNumber > Réduit `input` de `num_devices` en utilisant `reduction` à un seul périphérique.