TPUReplicateMetadata.Options

classe estática pública TPUReplicateMetadata.Options

Atributos opcionais para TPUReplicateMetadata

Métodos públicos

TPUReplicateMetadata.Options	allowSoftPlacement (booleano allowSoftPlacement)
TPUReplicateMetadata.Options	forma de computação (List<Long> forma de computação)
TPUReplicateMetadata.Options	deviceAssignment (List<Long> deviceAssignment)
TPUReplicateMetadata.Options	hostComputeCore (List<> hostComputeCore)
TPUReplicateMetadata.Options	numCoresPerReplica (long numCoresPerReplica)
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 a (objeto arg0)
Aula final<?>	getClass ()
int	código de hash ()
vazio final	notifique ()
vazio final	notifiqueTodos ()
Corda	toString ()
vazio final	espere (long arg0, int arg1)
vazio final	espere (arg0 longo)
vazio final	espere ()

Métodos públicos

public TPUReplicateMetadata.Options allowSoftPlacement (booleano allowSoftPlacement)

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

Parâmetros

computaçãoForma	DESCONTINUADA. Em vez disso, use num_cores_per_replica.

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)

public TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Parâmetros

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

public TPUReplicateMetadata.Options paddingMap (List<> paddingMap)

public TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

topologia pública TPUReplicateMetadata.Options ( topologia de string)

Parâmetros

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

public TPUReplicateMetadata.Options tpuCompileOptionsProto (String tpuCompileOptionsProto)

public TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

public TPUReplicateMetadata.Options useTpu (Boolean useTpu)

Parâmetros

use Tpu	Se o cálculo deve ser colocado na TPU.