OrdinalSelector

classe finale publique OrdinalSelector

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.

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

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

Méthodes héritées

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "TPUOrdinalSelector"

Méthodes publiques

Sortie publique < TInt32 > asOutput ()

Renvoie le handle symbolique du 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 d' un OrdinalSelector statique public (portée de la portée)

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

Paramètres
portée portée actuelle
Retour
  • une nouvelle instance d'OrdinalSelector

Sortie publique < TInt32 > deviceOrdinals ()

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