EnqueueTPUEmbeddingIntegerBatch

EnqueueTPUEmbeddingIntegerBatch public final class

Una operación que pone en cola una lista de tensores de lote de entrada en TPUEmbedding.

Clases anidadas

clase EnqueueTPUEmbeddingIntegerBatch.Options Los atributos opcionales para EnqueueTPUEmbeddingIntegerBatch

Métodos públicos

estática EnqueueTPUEmbeddingIntegerBatch
crear ( Alcance alcance, Iterable < operando <Integer >> lotes, operando <String> modeOverride, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación EnqueueTPUEmbeddingIntegerBatch.
estáticas EnqueueTPUEmbeddingIntegerBatch.Options
deviceOrdinal (Long deviceOrdinal)

Métodos heredados

Métodos públicos

public static EnqueueTPUEmbeddingIntegerBatch crear ( Alcance alcance, Iterable < operando <Integer >> lotes, operando <String> modeOverride, Opciones ... Opciones)

Método de fábrica para crear una clase que envuelva 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 cadena de entrada que anula el modo especificado en TPUEmbeddingConfiguration. Los valores admitidos son {'unspecified', 'inference', 'training', 'backward_pass_only'}. Cuando se establece en 'no especificado', 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 deviceOrdinal (Long deviceOrdinal)

Parámetros
deviceOrdinal El dispositivo de TPU que se utilizará. Debe ser> = 0 y menor que la cantidad de núcleos de TPU en la tarea en la que se coloca el nodo.