TPUOrdinalSelector

public final class TPUOrdinalSelector

Un selettore core TPU Op.

Questa operazione produce un set di core TPU (per il riscaldamento) o un singolo core TPU (per inferenza regolare) per eseguire il programma TPU. L'output viene utilizzato da TPUPartitionedCall.

Metodi pubblici

Uscita <Integer>
asOutput ()
Restituisce la maniglia simbolica di un tensore.
TPUOrdinalSelector statico
create ( ambito ambito)
Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione TPUOrdinalSelector.
Uscita <Integer>
deviceOrdinals ()
Un vettore 1 o più core TPU.

Metodi ereditati

Metodi pubblici

output pubblico <Integer> 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 TPUOrdinalSelector create ( ambito ambito)

Metodo Factory per creare una classe che avvolge una nuova operazione TPUOrdinalSelector.

Parametri
scopo ambito attuale
ritorna
  • una nuova istanza di TPUOrdinalSelector

output pubblico <Integer> deviceOrdinals ()

Un vettore 1 o più core TPU.