TPUReplicateMetadata

classe finale publique TPUReplicateMetadata

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 TPUReplicateMetadata.Options Attributs facultatifs pour TPUReplicateMetadata

Constantes

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

Méthodes publiques

statique TPUReplicateMetadata.Options
AllowSoftPlacement (booléen AllowSoftPlacement)
statique TPUReplicateMetadata.Options
calculShape (Liste<Long> calculShape)
TPUReplicateMetadata statique
créer (options Scope , Long numReplicas, Options... )
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUReplicateMetadata.
statique TPUReplicateMetadata.Options
DeviceAssignment (Liste <Long> DeviceAssignment)
statique TPUReplicateMetadata.Options
hostComputeCore (Liste<String> hostComputeCore)
statique TPUReplicateMetadata.Options
numCoresPerReplica (numCoresPerReplica long)
statique TPUReplicateMetadata.Options
paddingMap (Liste<String> paddingMap)
statique TPUReplicateMetadata.Options
stepMarkerLocation (chaîne stepMarkerLocation)
statique TPUReplicateMetadata.Options
topologie (topologie de chaîne)
statique TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning)
statique TPUReplicateMetadata.Options
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 statique TPUReplicateMetadata.OptionsallowSoftPlacement ( booléenallowSoftPlacement)

public static TPUReplicateMetadata.Options computationShape (List<Long> computationShape)

Paramètres
calculForme DÉCONSEILLÉ. Utilisez plutôt num_cores_per_replica.

public static TPUReplicateMetadata create ( Scope scope, Long numReplicas, Options... options)

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

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 TPUReplicateMetadata

public statique TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)

Paramètres
affectation des appareils L'affectation des appareils pour le calcul.

public statique TPUReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

public statique TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Paramètres
numCoresParReplica Nombre de cœurs par réplique. Utilisé pour le parallélisme des modèles.

public statique TPUReplicateMetadata.Options paddingMap (List<String> paddingMap)

public statique TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topologie publique statique TPUReplicateMetadata.Options (topologie String)

Paramètres
topologie TopologyProto indiquant la topologie de la tranche du pod TPU.

public statique TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning)

public statique TPUReplicateMetadata.Options useTpu (booléen useTpu)

Paramètres
utiliserTpu S'il faut placer le calcul sur le TPU.