Équilibrage de charge à tour de rôle sur les cœurs TPU.
Une opération d’équilibrage de charge qui alterne entre les cœurs TPU.
Cette opération effectue des round-robins entre les entiers dans [0, NumTPUCoresVisiblePerHost]. Il est utile pour s'interfacer avec les opérations TensorFlow qui prennent en entrée un cœur TPU sur lequel exécuter des calculs, tels que « TPUPartitionedCall ».
device_ordinal : un entier dans [0, NumTPUCoresVisiblePerHost].
Méthodes publiques
| Sortie <Entier> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
| TPURoundRobin statique | créer (portée d'application ) Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPURoundRobin. |
| Sortie <Entier> |
Méthodes héritées
Méthodes publiques
sortie publique <Integer> 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.
création de TPURoundRobin statique publique (portée d'application )
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPURoundRobin.
Paramètres
| portée | portée actuelle |
|---|
Retour
- une nouvelle instance de TPURoundRobin