TPUOrdinalSelector

classe finale publique TPUOrdinalSelector

Un sélecteur de noyau TPU Op.

Cette opération produit un ensemble de cœurs TPU (pour l'échauffement) ou un seul cœur TPU (pour une inférence régulière) sur lequel exécuter le programme TPU. La sortie est consommée par TPUPartitionedCall.

Méthodes publiques

Sortie <Entier>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
TPUOrdinalSelector statique
créer (portée d'application )
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUOrdinalSelector.
Sortie <Entier>
appareilsOrdinaux ()
Un vecteur d’un ou plusieurs cœurs TPU.

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.

public static TPUOrdinalSelector créer (portée de portée )

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

Paramètres
portée portée actuelle
Retour
  • une nouvelle instance de TPUOrdinalSelector

sortie publique <Integer> deviceOrdinals ()

Un vecteur d’un ou plusieurs cœurs TPU.