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 | |
| 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. |
|---|