Bilanciamento del carico round-robin sui core TPU.
Un'operazione di bilanciamento del carico che si svolge in round robin tra i core TPU.
Questa operazione esegue un round robin tra i numeri interi in [0, NumTPUCoresVisiblePerHost]. È utile per interfacciarsi con le operazioni TensorFlow che accettano come input un core TPU su cui eseguire calcoli, come "TPUPartitionedCall".
device_ordinal: un numero intero in [0, NumTPUCoresVisiblePerHost].
Metodi pubblici
Output <Intero> | comeuscita () Restituisce la maniglia simbolica di un tensore. |
TPU staticoRoundRobin | |
Output <Intero> |
Metodi ereditati
Metodi pubblici
output pubblico <intero> 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.
creazione TPURoundRobin pubblica statica (ambito ambito )
Metodo factory per creare una classe che racchiude una nuova operazione TPURoundRobin.
Parametri
scopo | ambito attuale |
---|
ritorna
- una nuova istanza di TPURoundRobin