Reçoit les données de déduplication (indices et pondérations) du noyau d'intégration.
Les données de déduplication sont un Tensor avec type=DT_VARIANT. Le tenseur lui-même est un tuple imbriqué XLA contenant N éléments (où N est le rapport entre le nombre d'intégrations et le nombre de cœurs de tenseur par puce TPU). Chaque élément du tuple imbriqué est un tuple de tenseurs de rang 1. Chaque tenseur contient soit des indices (DT_UINT32) pour la recherche d'intégration sur le TensorCore, soit des poids (DT_FLOAT) à appliquer à la sortie de l'opération de recherche d'intégration.
Méthodes publiques
Sortie <Objet> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
statique XlaRecvTPUEmbeddingDeduplicationData | |
Sortie <?> | sortir () |
Méthodes héritées
Méthodes publiques
sortie publique <Objet> 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 XlaRecvTPUEmbeddingDeduplicationData créer (portée d'étendue , configuration de chaîne)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération XlaRecvTPUEmbeddingDeduplicationData.
Paramètres
portée | portée actuelle |
---|---|
configuration | Protocole TPUEmbeddingConfiguration sérialisé. |
Retour
- une nouvelle instance de XlaRecvTPUEmbeddingDeduplicationData