ExecuteTPUEmbeddingPartitioner

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

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