EnqueueTPUEmbeddingIntegerBatch

classe final pública EnqueueTPUEmbeddingIntegerBatch

Uma operação que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.

Classes aninhadas

aula EnqueueTPUEmbeddingIntegerBatch.Options Atributos opcionais para EnqueueTPUEmbeddingIntegerBatch

Constantes

Corda OP_NAME O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Métodos Públicos

estático EnqueueTPUEmbeddingIntegerBatch
create ( Escopo , Iterable< Operando < TInt32 >> lote, Operando < TString > modeOverride, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação EnqueueTPUEmbeddingIntegerBatch.
estático EnqueueTPUEmbeddingIntegerBatch.Options
deviceOrdinal (dispositivo longo Ordinal)

Métodos herdados

Constantes

String final estática pública OP_NAME

O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Valor constante: "EnqueueTPUEmbeddingIntegerBatch"

Métodos Públicos

public static EnqueueTPUEmbeddingIntegerBatch create ( Scope scope, Iterable< Operand < TInt32 >> batch, Operand < TString > modeOverride, Options... options)

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

Parâmetros
escopo escopo atual
lote Uma lista de tensores 1D, um para cada tabela incorporada, contendo os índices nas tabelas.
modoSubstituir Uma entrada de string que substitui o modo especificado em TPUEmbeddingConfiguration. Os valores suportados são {'unspecified', 'inference', 'training', '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
dispositivoOrdinal O dispositivo TPU a ser usado. Deve ser >= 0 e menor que o número de núcleos de TPU na tarefa na qual o nó está colocado.