org.tensorflow.op.distribute

Clases

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