TPUReplicateMetadata.Options

classe statica pubblica TPUReplicateMetadata.Options

Attributi facoltativi per TPUReplicateMetadata

Metodi pubblici

TPUReplicateMetadata.Options	allowSoftPlacement (Boolean allowSoftPlacement)
TPUReplicateMetadata.Options	computationShape (List <Long> computationShape)
TPUReplicateMetadata.Options	deviceAssignment (List <Long> deviceAssignment)
TPUReplicateMetadata.Options	hostComputeCore (List <String> hostComputeCore)
TPUReplicateMetadata.Options	numCoresPerReplica (Long numCoresPerReplica)
TPUReplicateMetadata.Options	paddingMap (List <String> paddingMap)
TPUReplicateMetadata.Options	stepMarkerLocation (String stepMarkerLocation)
TPUReplicateMetadata.Options	topologia ( topologia di stringa)
TPUReplicateMetadata.Options	useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
TPUReplicateMetadata.Options	useTpu (booleano useTpu)

Metodi ereditati

Dalla classe java.lang.Object

booleano	è uguale a (Oggetto arg0)
classe finale <?>	getClass ()
int	hashCode ()
vuoto finale	notifica ()
vuoto finale	notifyAll ()
Corda	toString ()
vuoto finale	wait (long arg0, int arg1)
vuoto finale	aspetta (lungo arg0)
vuoto finale	aspetta ()

Metodi pubblici

public TPUReplicateMetadata.Options allowSoftPlacement (Boolean allowSoftPlacement)

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

Parametri

computationShape	DEPRECATO. Usa invece num_cores_per_replica.

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

Parametri

deviceAssignment	L'assegnazione di dispositivi per il calcolo.

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

public TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Parametri

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

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

public TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

topologia TPUReplicateMetadata.Options pubblica (topologia String)

Parametri

topologia	TopologyProto che indica la topologia dello slice del pod TPU.

public TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

public TPUReplicateMetadata.Options useTpu (Boolean useTpu)

Parametri

useTpu	Se posizionare il calcolo sul TPU.