classe finale publique ReplicateMetadata
Métadonnées indiquant comment le calcul TPU doit être répliqué.
Cette opération contient les métadonnées communes aux opérations d'un sous-graphe de calcul `tpu.replicate()`.
Classes imbriquées
| classe | ReplicateMetadata.Options | Attributs facultatifs pour ReplicateMetadata | |
Constantes
| Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
| ReplicateMetadata.Options statique | AllowSoftPlacement (booléen AllowSoftPlacement) |
| ReplicateMetadata.Options statique | calculShape (Liste<Long> calculShape) |
| RéplicationMétadonnées statiques | créer (options Scope , Long numReplicas, Options... ) Méthode d'usine pour créer une classe encapsulant une nouvelle opération ReplicateMetadata. |
| ReplicateMetadata.Options statique | DeviceAssignment (Liste <Long> DeviceAssignment) |
| ReplicateMetadata.Options statique | hostComputeCore (Liste<String> hostComputeCore) |
| ReplicateMetadata.Options statique | numCoresPerReplica (numCoresPerReplica long) |
| ReplicateMetadata.Options statique | paddingMap (Liste<String> paddingMap) |
| ReplicateMetadata.Options statique | stepMarkerLocation (chaîne stepMarkerLocation) |
| ReplicateMetadata.Options statique | topologie (topologie de chaîne) |
| ReplicateMetadata.Options statique | useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning) |
| ReplicateMetadata.Options statique | useTpu (booléen useTpu) |
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 : "TPUReplicateMetadata"
Méthodes publiques
public static ReplicateMetadata.Options computationShape (List<Long> computationShape)
Paramètres
| calculForme | DÉCONSEILLÉ. Utilisez plutôt num_cores_per_replica. |
|---|
public static ReplicateMetadata create ( Scope scope, Long numReplicas, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ReplicateMetadata.
Paramètres
| portée | portée actuelle |
|---|---|
| numRépliques | Nombre de répliques du calcul |
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de ReplicateMetadata
public statique ReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
Paramètres
| affectation des appareils | L'affectation des appareils pour le calcul. |
|---|
public statique ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
Paramètres
| numCoresParReplica | Nombre de cœurs par réplique. Utilisé pour le parallélisme des modèles. |
|---|
Topologie publique statique ReplicateMetadata.Options (topologie String)
Paramètres
| topologie | TopologyProto indiquant la topologie de la tranche du pod TPU. |
|---|
public static ReplicateMetadata.Options useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning)
public statique ReplicateMetadata.Options useTpu (booléen useTpu)
Paramètres
| utiliserTpu | S'il faut placer le calcul sur le TPU. |
|---|