clase final pública EnqueueTPUEmbeddingIntegerBatch
Una operación que pone en cola una lista de tensores por lotes de entrada para TPUEmbedding.
Clases anidadas
clase | EnqueueTPUEmbeddingIntegerBatch.Options | Atributos opcionales para EnqueueTPUEmbeddingIntegerBatch |
Métodos públicos
static EnqueueTPUEmbeddingIntegerBatch | crear ( Ámbito ámbito, Iterable< Operando <Entero>> lote, Operando <Cadena> modeOverride, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingIntegerBatch. |
static EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (dispositivo largo ordinal) |
Métodos Heredados
Métodos públicos
public static EnqueueTPUEmbeddingIntegerBatch create ( Scope scope, Iterable< Operand <Integer>> batch, Operand <String> modeOverride, Options... options)
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 en las tablas. |
modeOverride | Una entrada de cadena que invalida 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
public static EnqueueTPUEmbeddingIntegerBatch.Options dispositivo ordinal ( dispositivo largo ordinal)
Parámetros
dispositivoordinal | El dispositivo de 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. |
---|