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

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	tpuCompileOptionsProto (chaîne tpuCompileOptionsProto)
statique TPUReplicateMetadata.Options	useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning)
statique TPUReplicateMetadata.Options	useTpu (booléen useTpu)

Méthodes héritées

De la classe org.tensorflow.op.PrimitiveOp

booléen final	est égal (Obj objet)
entier final	Code de hachage ()
Opération	op () Renvoie l' Operation sous-jacente
Chaîne finale	àChaîne ()

De la classe java.lang.Object

booléen	est égal à (Objet arg0)
Classe finale<?>	obtenirClasse ()
int	Code de hachage ()
vide final	notifier ()
vide final	notifierTous ()
Chaîne	àChaîne ()
vide final	attendre (long arg0, int arg1)
vide final	attendez (long arg0)
vide final	attendez ()

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 tpuCompileOptionsProto (String tpuCompileOptionsProto)

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.