ConfigureTPUEmbeddingMemory

classe finale publique ConfigureTPUEmbeddingMemory

Une opération qui configure le logiciel TPUEmbedding sur un hôte.

Méthodes publiques

Sortie <Chaîne>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
statique ConfigureTPUEmbeddingMemory
créer ( Portée de portée , Opérande <String> commonConfig)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ConfigureTPUEmbeddingMemory.
Sortie <Chaîne>
mémoireConfig ()
Configuration de mémoire codée en chaîne contenant des métadonnées sur les allocations de mémoire réservées à TPUEmbedding.

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.

public static ConfigureTPUEmbeddingMemory create ( Scope scope, Operand <String> commonConfig)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération ConfigureTPUEmbeddingMemory.

Paramètres
portée portée actuelle
commonConfig Un proto CommonConfiguration codé en chaîne contenant des métadonnées sur la sortie du partitionneur TPUEmbedding et la taille HBM (en octets) requise pour le fonctionnement.
Retour
  • une nouvelle instance de ConfigureTPUEmbeddingMemory

sortie publique <String> memoryConfig ()

Configuration de mémoire codée en chaîne contenant des métadonnées sur les allocations de mémoire réservées à TPUEmbedding.