InfeedEnqueuePrelinearizedBuffer

classe finale pubblica InfeedEnqueuePrelinearizedBuffer

Un'operazione che accoda il buffer prelinearizzato nell'alimentazione TPU.

Classi nidificate

classe InfeedEnqueuePrelinearizedBuffer.Options Attributi facoltativi per InfeedEnqueuePrelinearizedBuffer

Metodi pubblici

InfeedEnqueuePrelinearizedBuffer statico
create ( ambito ambito , operando <?> input, opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione InfeedEnqueuePrelinearizedBuffer.
static InfeedEnqueuePrelinearizedBuffer.Options
dispositivoOrdinale (Dispositivo lungoOrdinale)

Metodi ereditati

Metodi pubblici

public static InfeedEnqueuePrelinearizedBuffer create ( ambito ambito , Operando <?> input, Opzioni... opzioni)

Metodo factory per creare una classe che racchiude una nuova operazione InfeedEnqueuePrelinearizedBuffer.

Parametri
scopo ambito attuale
ingresso Un tensore variante che rappresenta l'output linearizzato.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di InfeedEnqueuePrelinearizedBuffer

public static InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Long deviceOrdinal)

Parametri
dispositivoOrdinale Il dispositivo TPU da utilizzare. Dovrebbe essere -1 quando l'Operazione è in esecuzione su un dispositivo TPU e = 0 quando l'Operazione è in esecuzione sul dispositivo CPU.