ReplicateMetadata.Options

classe estática pública ReplicateMetadata.Options

Atributos opcionais para ReplicateMetadata

Métodos Públicos

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

Métodos herdados

Da classe java.lang.Object

boleano	é igual (objeto arg0)
aula final<?>	getClass ()
interno	código hash ()
vazio final	notificar ()
vazio final	notificar todos ()
Corda	para sequenciar ()
vazio final	espere (long arg0, int arg1)
vazio final	espere (arg0 longo)
vazio final	espere ()

Métodos Públicos

public ReplicateMetadata.Options permitirSoftPlacement (booleano permitirSoftPlacement)

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

Parâmetros

computaçãoForma	DESCONTINUADA. Use num_cores_per_replica em vez disso.

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

Parâmetros

atribuição de dispositivo	A atribuição de dispositivos para a computação.

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

público ReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica longo)

Parâmetros

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

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

public ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topologia pública ReplicateMetadata.Options (topologia String)

Parâmetros

topologia	TopologyProto indicando a topologia da fatia do pod TPU.

public ReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

public ReplicateMetadata.Options useTpu (booleano useTpu)

Parâmetros

usarTpu	Se o cálculo deve ser colocado na TPU.