classe final pública EnqueueTPUEmbeddingIntegerBatch
Um op que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.
Classes aninhadas
| aula | EnqueueTPUEmbeddingIntegerBatch.Options | Atributos opcionais para EnqueueTPUEmbeddingIntegerBatch | |
Métodos públicos
| static EnqueueTPUEmbeddingIntegerBatch | |
| static EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (Longo deviceOrdinal) |
Métodos Herdados
Métodos públicos
public static EnqueueTPUEmbeddingIntegerBatch create ( Escopo do escopo, Iterable< Operando <Integer>> lote, Operando <String> modeOverride, Opções... 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 no TPUEmbeddingConfiguration. Os valores suportados são {'unspecified', 'inference', 'training', 'backward_pass_only'}. Quando definido como 'unspecified', 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ó é colocado. |
|---|