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

Constantes

Fragmento OP_NAME O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Métodos Públicos

estática EnqueueTPUEmbeddingIntegerBatch
criar ( Scope escopo, Iterable < Operando < TInt32 >> lote, operando < TString > 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

Constantes

nome_op final String public static

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

Valor constante: "EnqueueTPUEmbeddingIntegerBatch"

Métodos Públicos

public static EnqueueTPUEmbeddingIntegerBatch criar ( Scope escopo, Iterable < Operando < TInt32 >> lote, operando < TString > 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.