classe finale publique ExecuteTPUEmbeddingPartitioner
Une opération qui exécute le partitionneur TPUEmbedding sur la configuration centrale
périphérique et calcule la taille HBM (en octets) requise pour l’opération TPUEmbedding.
Méthodes publiques
| Sortie <Chaîne> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
| Sortie <Chaîne> | configuration commune () Un protocole de configuration commun codé sous forme de chaîne contenant des métadonnées sur la sortie du partitionneur TPUEmbedding et la taille HBM (en octets) requise pour le fonctionnement. |
| ExecuteTPUEmbeddingPartitioner statique |
Méthodes héritées
Méthodes publiques
sortie publique <String> asOutput ()
Renvoie le handle symbolique d'un tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
sortie publique <String> commonConfig ()
Un protocole de configuration commun codé sous forme de chaîne contenant des métadonnées sur la sortie du partitionneur TPUEmbedding et la taille HBM (en octets) requise pour le fonctionnement.
public static ExecuteTPUEmbeddingPartitioner créer (portée de portée , configuration de chaîne)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ExecuteTPUEmbeddingPartitioner.
Paramètres
| portée | portée actuelle |
|---|---|
| configuration | Un proto TPUEmbeddingConfiguration sérialisé en une chaîne, décrivant la configuration TPUEmbedding souhaitée. |
Retour
- une nouvelle instance de ExecuteTPUEmbeddingPartitioner