EnqueueTPUEmbeddingIntegerBatch

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

Métodos públicos

Enqueue estáticoTPUEmbeddingIntegerBatch
crear (alcance alcance , Iterable< Operando <Integer>> lote, Operando <Cadena> modeOverride, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación EnqueueTPUEmbeddingIntegerBatch.
EnqueueTPUEmbeddingIntegerBatch.Options estático
dispositivoOrdinal (dispositivo largoOrdinal)

Métodos heredados

Métodos públicos

public static EnqueueTPUEmbeddingIntegerBatch create (alcance de alcance , iterable < operando <integer>> lote, operando <cadena> 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.