TPUReplicateMetadata

classe finale pubblica TPUReplicateMetadata

Metadati che indicano come replicare il calcolo della TPU.

Questa operazione contiene i metadati comuni alle operazioni di un sottografo di calcolo `tpu.replicate()`.

Classi nidificate

classe

TPUReplicateMetadata.Options

Attributi facoltativi per TPUReplicateMetadata

Metodi pubblici

TPUReplicateMetadata.Options statico	allowSoftPlacement (allowSoftPlacement booleano)
TPUReplicateMetadata.Options statico	computationShape (Lista<Long> computationShape)
TPUReplicateMetadata statici	crea (ambito ambito , numRepliche lunghe, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione TPUReplicateMetadata.
TPUReplicateMetadata.Options statico	assegnazione dispositivo (Elenco<lungo> assegnazione dispositivo)
TPUReplicateMetadata.Options statico	hostComputeCore (Lista<String> hostComputeCore)
TPUReplicateMetadata.Options statico	numCoresPerReplica (numCoresPerReplica lungo)
TPUReplicateMetadata.Options statico	imbottituraMap (Lista<String> imbottituraMap)
TPUReplicateMetadata.Options statico	stepMarkerLocation (Stringa stepMarkerLocation)
TPUReplicateMetadata.Options statico	topologia (topologia di stringa)
TPUReplicateMetadata.Options statico	useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)
TPUReplicateMetadata.Options statico	useTpu (useTpu booleano)

Metodi ereditati

Dalla classe org.tensorflow.op.PrimitiveOp

booleano finale	è uguale a (Oggetto oggetto)
finale int	codice hash ()
Operazione	operazione () Restituisce l' Operation sottostante
stringa finale	accordare ()

Dalla classe java.lang.Object

booleano	è uguale a (Oggetto arg0)
Classe finale<?>	getClass ()
int	codice hash ()
vuoto finale	notificare ()
vuoto finale	notificaTutti ()
Corda	accordare ()
vuoto finale	attendere (lungo arg0, int arg1)
vuoto finale	aspetta (lungo arg0)
vuoto finale	Aspettare ()

Metodi pubblici

pubblico statico TPUReplicateMetadata.OptionsallowSoftPlacement ( allowSoftPlacement booleano)

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

Parametri

calcoloShape	DEPRECATO. Utilizzare invece num_cores_per_replica.

creazione pubblica statica TPUReplicateMetadata (ambito ambito , numReplicas lunghi, Opzioni... opzioni)

Metodo factory per creare una classe che racchiude una nuova operazione TPUReplicateMetadata.

Parametri

scopo	ambito attuale
numRepliche	Numero di repliche del calcolo
opzioni	trasporta valori di attributi opzionali

ritorna

• una nuova istanza di TPUReplicateMetadata

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

Parametri

dispositivoAssegnazione	L'assegnazione dei dispositivi per il calcolo.

pubblico statico TPUReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

pubblico statico TPUReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica lungo)

Parametri

numCoresPerReplica	Numero di core per replica. Utilizzato per il parallelismo del modello.

pubblico statico TPUReplicateMetadata.Options imbottituraMap (List<String> imbottituraMap)

pubblico statico TPUReplicateMetadata.Options stepMarkerLocation (Stringa stepMarkerLocation)

Topologia statica pubblica TPUReplicateMetadata.Options (topologia stringa)

Parametri

topologia	TopologyProto che indica la topologia della sezione del pod in TPU.

pubblico statico TPUReplicateMetadata.Options useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)

pubblico statico TPUReplicateMetadata.Options useTpu (useTpu booleano)

Parametri

utilizzareTpu	Se posizionare il calcolo sul TPU.