TPUReplicateMetadata.Options

classe statique publique TPUReplicateMetadata.Options

Attributs facultatifs pour TPUReplicateMetadata

Méthodes publiques

TPUReplicateMetadata.Options	AllowSoftPlacement (booléen AllowSoftPlacement)
TPUReplicateMetadata.Options	calculShape (Liste<Long> calculShape)
TPUReplicateMetadata.Options	DeviceAssignment (Liste <Long> DeviceAssignment)
TPUReplicateMetadata.Options	hostComputeCore (Liste<String> hostComputeCore)
TPUReplicateMetadata.Options	numCoresPerReplica (numCoresPerReplica long)
TPUReplicateMetadata.Options	paddingMap (Liste<String> paddingMap)
TPUReplicateMetadata.Options	stepMarkerLocation (chaîne stepMarkerLocation)
TPUReplicateMetadata.Options	topologie (topologie de chaîne)
TPUReplicateMetadata.Options	useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning)
TPUReplicateMetadata.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 TPUReplicateMetadata.OptionsallowSoftPlacement ( booléenallowSoftPlacement)

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

Paramètres

calculForme	DÉCONSEILLÉ. Utilisez plutôt num_cores_per_replica.

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

Paramètres

affectation des appareils	L'affectation des appareils pour le calcul.

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

public TPUReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica long)

Paramètres

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

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

public TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topologie publique TPUReplicateMetadata.Options (topologie String)

Paramètres

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

public TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning)

public TPUReplicateMetadata.Options useTpu (booléen useTpu)

Paramètres

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