clase final pública EnqueueTPUEmbeddingIntegerBatch
Una operación que pone en cola una lista de tensores por lotes de entrada en TPUEmbedding.
Clases anidadas
| clase | Poner en colaTPUEmbeddingIntegerBatch.Options | Atributos opcionales para EnqueueTPUEmbeddingIntegerBatch | |
Constantes
| Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
| Enqueue estáticoTPUEmbeddingIntegerBatch | |
| EnqueueTPUEmbeddingIntegerBatch.Options estático | dispositivoOrdinal (dispositivo largoOrdinal) |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Valor constante: "EnqueueTPUEmbeddingIntegerBatch"
Métodos públicos
public static EnqueueTPUEmbeddingIntegerBatch create (Ámbito de alcance , Iterable < Operando < TInt32 >> lote, Operando < TString > modeOverride, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingIntegerBatch.
Parámetros
| alcance | alcance actual |
|---|---|
| lote | Una lista de tensores 1D, uno para cada tabla de incrustación, que contiene los índices de las tablas. |
| anulación de modo | Una entrada de cadena que anula el modo especificado en TPUEmbeddingConfiguration. Los valores admitidos son {'unspecified', 'inference', 'training', 'backward_pass_only'}. Cuando se establece en 'sin especificar', se usa el modo establecido en TPUEmbeddingConfiguration; de lo contrario, se usa mode_override. |
| opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de EnqueueTPUEmbeddingIntegerBatch
público estático EnqueueTPUEmbeddingIntegerBatch.Options dispositivoOrdinal (dispositivo largo Ordinal)
Parámetros
| dispositivoOrdinal | El dispositivo TPU a utilizar. Debe ser >= 0 y menor que la cantidad de núcleos de TPU en la tarea en la que se coloca el nodo. |
|---|