NcclAllReduce

public final class NcclAllReduce

Emette un tensore contenente la riduzione su tutti i tensori di input.

Emette un tensore contenente la riduzione su tutti i tensori di input passati agli operatori all'interno dello stesso `shared_name.

Il grafico dovrebbe essere costruito in modo che se un op viene eseguito con il valore shared_name "c", allora gli op "num_devices" verranno eseguiti con il valore shared_name "c". In caso contrario, l'esecuzione del grafico non verrà completata.

input: l'input per i dati di riduzione: il valore della riduzione su tutti i dispositivi `num_devices`. riduzione: l'operazione di riduzione da eseguire. num_devices: il numero di dispositivi che partecipano a questa riduzione. shared_name: identificatore condiviso tra le operazioni della stessa riduzione.

Metodi pubblici

Uscita <T>
asOutput ()
Restituisce la maniglia simbolica di un tensore.
static <T extends Number> NcclAllReduce <T>
create ( ambito ambito, operando <T> input, riduzione stringhe, long numDevices, string sharedName)
Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione NcclAllReduce.
Uscita <T>
dati ()

Metodi ereditati

Metodi pubblici

output pubblico <T> asOutput ()

Restituisce la maniglia simbolica di un tensore.

Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.

public static NcclAllReduce <T> create ( ambito ambito, operando <T> input, riduzione stringhe, numero lungo dispositivi, stringa sharedName)

Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione NcclAllReduce.

Parametri
scopo ambito attuale
ritorna
  • una nuova istanza di NcclAllReduce

output pubblico <T> dati ()