EnqueueTPUEmbeddingIntegerBatch

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

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

EnqueueTPUEmbeddingIntegerBatch statique
créer ( Scope scope, Iterable< Operand < TInt32 >> batch, Operand < TString > 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

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "EnqueueTPUEmbeddingIntegerBatch"

Méthodes publiques

public static EnqueueTPUEmbeddingIntegerBatch create ( Scope scope, Iterable< Operand < TInt32 >> batch, Operand < TString > 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é.