classe statique publique ReplicateMetadata.Options
Attributs facultatifs pour ReplicateMetadata
Méthodes publiques
| ReplicateMetadata.Options | AllowSoftPlacement (booléen AllowSoftPlacement) |
| ReplicateMetadata.Options | calculShape (Liste<Long> calculShape) |
| ReplicateMetadata.Options | DeviceAssignment (Liste <Long> DeviceAssignment) |
| ReplicateMetadata.Options | hostComputeCore (Liste<String> hostComputeCore) |
| ReplicateMetadata.Options | numCoresPerReplica (numCoresPerReplica long) |
| ReplicateMetadata.Options | paddingMap (Liste<String> paddingMap) |
| ReplicateMetadata.Options | stepMarkerLocation (chaîne stepMarkerLocation) |
| ReplicateMetadata.Options | topologie (topologie de chaîne) |
| ReplicateMetadata.Options | useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning) |
| ReplicateMetadata.Options | useTpu (booléen useTpu) |
Méthodes héritées
Méthodes publiques
public ReplicateMetadata.Options computationShape (List<Long> computationShape)
Paramètres
| calculForme | DÉCONSEILLÉ. Utilisez plutôt num_cores_per_replica. |
|---|
public ReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
Paramètres
| affectation des appareils | L'affectation des appareils pour le calcul. |
|---|
public ReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica long)
Paramètres
| numCoresParReplica | Nombre de cœurs par réplique. Utilisé pour le parallélisme des modèles. |
|---|
Topologie publique ReplicateMetadata.Options (topologie String)
Paramètres
| topologie | TopologyProto indiquant la topologie de la tranche du pod TPU. |
|---|
public ReplicateMetadata.Options useTpu (booléen useTpu)
Paramètres
| utiliserTpu | S'il faut placer le calcul sur le TPU. |
|---|