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. |