TPURoundRobin

classe finale publique TPURoundRobin

É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

sortie publique <Integer> deviceOrdinal ()