EnqueueTPUEmbeddingIntegerBatch

EnqueueTPUEmbeddingIntegerBatch classe final pública

Um op que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.

Classes aninhadas

classe

EnqueueTPUEmbeddingIntegerBatch.Options

Atributos opcionais para EnqueueTPUEmbeddingIntegerBatch

Métodos Públicos

estática EnqueueTPUEmbeddingIntegerBatch	criar ( Scope escopo, Iterable < Operando <Integer >> lote, operando <String> modeOverride, Options ... Opções) Método de fábrica para criar uma classe que envolve uma nova operação EnqueueTPUEmbeddingIntegerBatch.
estáticos EnqueueTPUEmbeddingIntegerBatch.Options	deviceOrdinal (Long deviceOrdinal)

Métodos herdados

De classe org.tensorflow.op.PrimitiveOp

final booleano	é igual a (obj Object)
final int	hashCode ()
Operação	op () Retorna o subjacente Operation
corda final	toString ()

Da classe java.lang.Object

boleano	equals (arg0 Object)
aula final <?>	getClass ()
int	hashCode ()
vazio final	notificar ()
vazio final	notifyAll ()
Fragmento	toString ()
vazio final	wait (longa arg0, int arg1)
vazio final	wait (arg0 de comprimento)
vazio final	wait ()

Métodos Públicos

public static EnqueueTPUEmbeddingIntegerBatch criar ( Scope escopo, Iterable < Operando <Integer >> lote, operando <String> modeOverride, Options ... Opções)

Método de fábrica para criar uma classe que envolve uma nova operação EnqueueTPUEmbeddingIntegerBatch.

Parâmetros

alcance	escopo atual
lote	Uma lista de tensores 1D, um para cada tabela de incorporação, contendo os índices nas tabelas.
modeOverride	Uma entrada de string que substitui o modo especificado na TPUEmbeddingConfiguration. Os valores suportados são {'não especificado', 'inferência', 'treinamento', 'backward_pass_only'}. Quando definido como 'não especificado', o modo definido em TPUEmbeddingConfiguration é usado, caso contrário, mode_override é usado.
opções	carrega valores de atributos opcionais

Devoluções

• uma nova instância de EnqueueTPUEmbeddingIntegerBatch

public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)

Parâmetros

deviceOrdinal	O dispositivo TPU a ser usado. Deve ser> = 0 e menor que o número de núcleos de TPU na tarefa em que o nó está colocado.