Aviso: esta API está obsoleta e será removida em uma versão futura do TensorFlow depois que a substituição estiver estável.

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

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.