EnqueueTPUEmbeddingIntegerBatch

classe finale pubblica EnqueueTPUEmbeddingIntegerBatch

Un'operazione che accoda un elenco di tensori batch di input a TPUEmbedding.

Classi nidificate

classe EnqueueTPUEmbeddingIntegerBatch.Options Attributi facoltativi per EnqueueTPUEmbeddingIntegerBatch

Costanti

Corda OP_NAME Il nome di questa operazione, come noto al motore principale di TensorFlow

Metodi pubblici

EnqueueTPUEmbeddingIntegerBatch statico
create ( Scope scope, Iterable< Operand < TInt32 >> batch, Operand < TString > modeOverride, Options... options)
Metodo factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingIntegerBatch.
EnqueueTPUEmbeddingIntegerBatch.Options statico
dispositivoOrdinale (Dispositivo lungoOrdinale)

Metodi ereditati

Costanti

Stringa finale statica pubblica OP_NAME

Il nome di questa operazione, come noto al motore principale di TensorFlow

Valore costante: "EnqueueTPUEmbeddingIntegerBatch"

Metodi pubblici

public static EnqueueTPUEmbeddingIntegerBatch create ( scope scope, Iterable< Operand < TInt32 >> batch, Operand < TString > modeOverride, Options... options)

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

Parametri
scopo ambito attuale
lotto Un elenco di tensori 1D, uno per ciascuna tabella di incorporamento, contenente gli indici nelle tabelle.
modeOverride Un input di tipo stringa che sovrascrive la modalità specificata in TPUEmbeddingConfiguration. I valori supportati sono {'unspecified', 'inference', 'training', 'backward_pass_only'}. Se impostato su "non specificato", viene utilizzata la modalità impostata in TPUEmbeddingConfiguration, altrimenti viene utilizzato mode_override.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di EnqueueTPUEmbeddingIntegerBatch

public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)

Parametri
dispositivoOrdinale Il dispositivo TPU da utilizzare. Dovrebbe essere >= 0 e inferiore al numero di core TPU nell'attività su cui è posizionato il nodo.