EnqueueTPUEmbeddingIntegerBatch

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

De la clase org.tensorflow.op.PrimitiveOp

booleano final	es igual a (Objeto obj)
int final	código hash ()
Operación	op () Devuelve la Operation subyacente
cadena final	a la cadena ()

De la clase java.lang.Object

booleano	es igual a (Objeto arg0)
clase final<?>	obtenerClase ()
En t	código hash ()
vacío final	notificar ()
vacío final	notificar a todos ()
Cuerda	a la cadena ()
vacío final	espera (largo arg0, int arg1)
vacío final	espera (largo arg0)
vacío final	espera ()

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.