org.tensorflow.op.distribute

Clases

NcclAllReduce <T se extiende TNumber > Genera un tensor que contiene la reducción en todos los tensores de entrada.
NcclBroadcast <T se extiende TNumber > Envía "entrada" a todos los dispositivos que están conectados a la salida.
NcclReduce <T se extiende TNumber > Reduce la `entrada` de` num_devices` usando `reducción` a un solo dispositivo.