ReplicateMetadata.Options

ReplicateMetadata.Options classe estática pública

Atributos opcionais para ReplicateMetadata

Métodos Públicos

ReplicateMetadata.Options	allowSoftPlacement (booleano allowSoftPlacement)
ReplicateMetadata.Options	computationShape (List <longo> computationShape)
ReplicateMetadata.Options	deviceAssignment (List <longo> deviceAssignment)
ReplicateMetadata.Options	hostComputeCore (List <String> hostComputeCore)
ReplicateMetadata.Options	numCoresPerReplica (Long numCoresPerReplica)
ReplicateMetadata.Options	paddingMap (List <String> paddingMap)
ReplicateMetadata.Options	stepMarkerLocation (String stepMarkerLocation)
ReplicateMetadata.Options	topologia (topologia String)
ReplicateMetadata.Options	useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
ReplicateMetadata.Options	useTpu (booleano useTpu)

Métodos herdados

Da classe java.lang.Object

boleano	equals (arg0 Object)
aula final <?>	getClass ()
int	hashCode ()
vazio final	notificar ()
vazio final	notifyAll ()
Fragmento	toString ()
vazio final	wait (longa arg0, int arg1)
vazio final	wait (arg0 de comprimento)
vazio final	wait ()

Métodos Públicos

pública ReplicateMetadata.Options allowSoftPlacement (Boolean allowSoftPlacement)

pública ReplicateMetadata.Options computationShape (List <longo> computationShape)

Parâmetros

computationShape	DESCONTINUADA. Em vez disso, use num_cores_per_replica.

pública ReplicateMetadata.Options deviceAssignment (List <longo> deviceAssignment)

Parâmetros

deviceAssignment	A atribuição de dispositivos para o cálculo.

pública ReplicateMetadata.Options hostComputeCore (List <String> hostComputeCore)

pública ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Parâmetros

numCoresPerReplica	Número de núcleos por réplica. Usado para paralelismo de modelo.

pública ReplicateMetadata.Options paddingMap (List <String> paddingMap)

pública ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

pública ReplicateMetadata.Options topologia (topologia String)

Parâmetros

topologia	TopologyProto indicando a topologia da fatia do pod de TPU.

pública ReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

pública ReplicateMetadata.Options useTpu (Boolean useTpu)

Parâmetros

useTpu	Se deve colocar o cálculo na TPU.