TPUReplicateMetadata.Options

classe estática pública TPUReplicateMetadata.Options

Atributos opcionais para TPUReplicateMetadata

Métodos Públicos

TPUReplicateMetadata.Options	permitirSoftPlacement (booleano permitirSoftPlacement)
TPUReplicateMetadata.Options	computationShape (List<Long> computationShape)
TPUReplicateMetadata.Options	deviceAssignment (Lista<Long> deviceAssignment)
TPUReplicateMetadata.Options	hostComputeCore (List<String> hostComputeCore)
TPUReplicateMetadata.Options	numCoresPerReplica (numCoresPerReplica longo)
TPUReplicateMetadata.Options	paddingMap (List<String> paddingMap)
TPUReplicateMetadata.Options	stepMarkerLocation (String stepMarkerLocation)
TPUReplicateMetadata.Options	topologia (topologia de string)
TPUReplicateMetadata.Options	tpuCompileOptionsProto (String tpuCompileOptionsProto)
TPUReplicateMetadata.Options	useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
TPUReplicateMetadata.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 TPUReplicateMetadata.Options permitirSoftPlacement (booleano permitirSoftPlacement)

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

Parâmetros

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

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

Parâmetros

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

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

público TPUReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica longo)

Parâmetros

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

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

public TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

topologia pública TPUReplicateMetadata.Options (topologia String)

Parâmetros

topologia	TopologyProto indicando a topologia da fatia do pod TPU.

public TPUReplicateMetadata.Options tpuCompileOptionsProto (String tpuCompileOptionsProto)

public TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

public TPUReplicateMetadata.Options useTpu (booleano useTpu)

Parâmetros

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