TPUReplicateMetadata.Options

classe statica pubblica TPUReplicateMetadata.Options

Attributi facoltativi per TPUReplicateMetadata

Metodi pubblici

TPUReplicateMetadata.Options	allowSoftPlacement (allowSoftPlacement booleano)
TPUReplicateMetadata.Options	computationShape (Lista<Long> computationShape)
TPUReplicateMetadata.Options	assegnazione dispositivo (Elenco<lungo> assegnazione dispositivo)
TPUReplicateMetadata.Options	hostComputeCore (Lista<String> hostComputeCore)
TPUReplicateMetadata.Options	numCoresPerReplica (numCoresPerReplica lungo)
TPUReplicateMetadata.Options	imbottituraMap (Lista<String> imbottituraMap)
TPUReplicateMetadata.Options	stepMarkerLocation (Stringa stepMarkerLocation)
TPUReplicateMetadata.Options	topologia (topologia di stringa)
TPUReplicateMetadata.Options	useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)
TPUReplicateMetadata.Options	useTpu (useTpu booleano)

Metodi ereditati

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

public TPUReplicateMetadata.OptionsallowSoftPlacement ( allowSoftPlacement booleano)

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

Parametri

calcoloShape	DEPRECATO. Utilizzare invece num_cores_per_replica.

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

Parametri

dispositivoAssegnazione	L'assegnazione dei dispositivi per il calcolo.

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

public TPUReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica lungo)

Parametri

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

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

public TPUReplicateMetadata.Options stepMarkerLocation (Stringa stepMarkerLocation)

topologia pubblica TPUReplicateMetadata.Options (topologia stringa)

Parametri

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

public TPUReplicateMetadata.Options useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)

public TPUReplicateMetadata.Options useTpu (useTpu booleano)

Parametri

utilizzareTpu	Se posizionare il calcolo sul TPU.