TPURoundRobin

classe finale pubblica TPURoundRobin

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
creare ( ambito ambito )
Metodo factory per creare una classe che racchiude una nuova operazione TPURoundRobin.
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

public Output <Integer> deviceOrdinal ()