ReplicateMetadata

classe finale publique ReplicateMetadata

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

ReplicateMetadata.Options

Attributs facultatifs pour ReplicateMetadata

Constantes

Chaîne

OP_NAME

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

Méthodes publiques

ReplicateMetadata.Options statique	AllowSoftPlacement (booléen AllowSoftPlacement)
ReplicateMetadata.Options statique	calculShape (Liste<Long> calculShape)
RéplicationMétadonnées statiques	créer (options Scope , Long numReplicas, Options... ) Méthode d'usine pour créer une classe encapsulant une nouvelle opération ReplicateMetadata.
ReplicateMetadata.Options statique	DeviceAssignment (Liste <Long> DeviceAssignment)
ReplicateMetadata.Options statique	hostComputeCore (Liste<String> hostComputeCore)
ReplicateMetadata.Options statique	numCoresPerReplica (numCoresPerReplica long)
ReplicateMetadata.Options statique	paddingMap (Liste<String> paddingMap)
ReplicateMetadata.Options statique	stepMarkerLocation (chaîne stepMarkerLocation)
ReplicateMetadata.Options statique	topologie (topologie de chaîne)
ReplicateMetadata.Options statique	useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning)
ReplicateMetadata.Options statique	useTpu (booléen useTpu)

Méthodes héritées

De la classe org.tensorflow.op.RawOp

booléen final	est égal (Obj objet)
entier final	Code de hachage ()
Opération	op () Renvoie cette unité de calcul sous la forme d'une seule Operation .
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 ()

Depuis l'interface org.tensorflow.op.Op

résuméExécutionEnvironnement	env () Renvoie l'environnement d'exécution dans lequel cette opération a été créée.
opération abstraite	op () Renvoie cette unité de calcul sous la forme d'une seule Operation .

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

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

Paramètres

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

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

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

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 ReplicateMetadata

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

Paramètres

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

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

public statique ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Paramètres

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

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

public statique ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topologie publique statique ReplicateMetadata.Options (topologie String)

Paramètres

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

public static ReplicateMetadata.Options useSpmdForXlaPartitioning (booléen useSpmdForXlaPartitioning)

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

Paramètres

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