InfeedEnqueuePrelinearizedBuffer

public final class InfeedEnqueuePrelinearizedBuffer

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

Classi annidate

classe InfeedEnqueuePrelinearizedBuffer.Options Attributi facoltativi per InfeedEnqueuePrelinearizedBuffer

Metodi pubblici

static InfeedEnqueuePrelinearizedBuffer
create ( ambito ambito, operando <?> input, opzioni ... opzioni)
Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione InfeedEnqueuePrelinearizedBuffer.
static InfeedEnqueuePrelinearizedBuffer.Options
deviceOrdinal (dispositivo lungoOrdinal)

Metodi ereditati

Metodi pubblici

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

Metodo Factory per creare una classe che esegue il wrapping di 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
deviceOrdinal Il dispositivo TPU da utilizzare. Dovrebbe essere -1 quando Op è in esecuzione su un dispositivo TPU e = 0 quando Op è in esecuzione sul dispositivo CPU.