TPUReplicateMetadata.Options

TPUReplicateMetadata.Options classe estática pública

Atributos opcionais para TPUReplicateMetadata

Métodos Públicos

TPUReplicateMetadata.Options	allowSoftPlacement (booleano allowSoftPlacement)
TPUReplicateMetadata.Options	computationShape (List <longo> computationShape)
TPUReplicateMetadata.Options	deviceAssignment (List <longo> 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 String)
TPUReplicateMetadata.Options	useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
TPUReplicateMetadata.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 TPUReplicateMetadata.Options allowSoftPlacement (Boolean allowSoftPlacement)

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

Parâmetros

computationShape	DESCONTINUADA. Em vez disso, use num_cores_per_replica.

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

Parâmetros

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

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

pública TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Parâmetros

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

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

pública TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

pública TPUReplicateMetadata.Options topologia (topologia String)

Parâmetros

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

pública TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

pública TPUReplicateMetadata.Options useTpu (Boolean useTpu)

Parâmetros

useTpu	Se deve colocar o cálculo na TPU.