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. |
---|