ReplicateMetadata.Options

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

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 ReplicateMetadata.OptionsallowSoftPlacement ( booléenallowSoftPlacement)

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 hostComputeCore (List<String> hostComputeCore)

public ReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica long)

Paramètres

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

public ReplicateMetadata.Options paddingMap (List<String> paddingMap)

public ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topologie publique ReplicateMetadata.Options (topologie String)

Paramètres

topologie	TopologyProto indiquant la topologie de la tranche du pod TPU.

public ReplicateMetadata.Options useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning)

public ReplicateMetadata.Options useTpu (booléen useTpu)

Paramètres

utiliserTpu	S'il faut placer le calcul sur le TPU.