classe finale publique EnqueueTPUEmbeddingIntegerBatch
Une opération qui met en file d'attente une liste de tenseurs de lots d'entrée dans TPUEmbedding.
Classes imbriquées
classe | Mettre en file d'attenteTPUEmbeddingIntegerBatch.Options | Attributs facultatifs pour EnqueueTPUEmbeddingIntegerBatch |
Méthodes publiques
EnqueueTPUEmbeddingIntegerBatch statique | créer ( Scope scope, Iterable< Operand <Integer>> batch, Operand <String> modeOverride, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération EnqueueTPUEmbeddingIntegerBatch. |
EnqueueTPUEmbeddingIntegerBatch.Options statique | appareilOrdinal (appareilOrdinal long) |
Méthodes héritées
Méthodes publiques
public static EnqueueTPUEmbeddingIntegerBatch create ( Scope scope, Iterable< Operand <Integer>> batch, Operand <String> modeOverride, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération EnqueueTPUEmbeddingIntegerBatch.
Paramètres
portée | portée actuelle |
---|---|
lot | Une liste de tenseurs 1D, un pour chaque table d'intégration, contenant les indices dans les tables. |
modeOverride | Entrée de chaîne qui remplace le mode spécifié dans TPUEmbeddingConfiguration. Les valeurs prises en charge sont {'unspecified', 'inference', 'training', 'backward_pass_only'}. Lorsqu'il est défini sur « non spécifié », le mode défini dans TPUEmbeddingConfiguration est utilisé, sinon mode_override est utilisé. |
choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de EnqueueTPUEmbeddingIntegerBatch
public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
Paramètres
appareilOrdinal | Le périphérique TPU à utiliser. Doit être >= 0 et inférieur au nombre de cœurs TPU dans la tâche sur laquelle le nœud est placé. |
---|