TPUReplicateMetadata

classe final pública TPUReplicateMetadata

Metadados que indicam como o cálculo da TPU deve ser replicado.

Esta operação contém os metadados comuns às operações de um subgráfico de computação `tpu.replicate()`.

Classes aninhadas

aula

TPUReplicateMetadata.Options

Atributos opcionais para TPUReplicateMetadata

Métodos Públicos

TPUReplicateMetadata.Options estático	permitirSoftPlacement (booleano permitirSoftPlacement)
TPUReplicateMetadata.Options estático	computationShape (List<Long> computationShape)
TPUReplicateMetadata estático	create ( Escopo , Long numReplicas, Opções... opções) Método de fábrica para criar uma classe que envolve uma nova operação TPUReplicateMetadata.
TPUReplicateMetadata.Options estático	deviceAssignment (Lista<Long> deviceAssignment)
TPUReplicateMetadata.Options estático	hostComputeCore (List<String> hostComputeCore)
TPUReplicateMetadata.Options estático	numCoresPerReplica (numCoresPerReplica longo)
TPUReplicateMetadata.Options estático	paddingMap (List<String> paddingMap)
TPUReplicateMetadata.Options estático	stepMarkerLocation (String stepMarkerLocation)
TPUReplicateMetadata.Options estático	topologia (topologia de string)
TPUReplicateMetadata.Options estático	tpuCompileOptionsProto (String tpuCompileOptionsProto)
TPUReplicateMetadata.Options estático	useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
TPUReplicateMetadata.Options estático	useTpu (booleano useTpu)

Métodos herdados

Da classe org.tensorflow.op.PrimitiveOp

booleano final	é igual (objeto obj)
int final	código hash ()
Operação	op () Retorna a Operation subjacente
sequência final	para sequenciar ()

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

público estático TPUReplicateMetadata.Options permitirSoftPlacement (booleano permitirSoftPlacement)

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

Parâmetros

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

public static TPUReplicateMetadata create ( Escopo escopo, Long numReplicas, Opções... opções)

Método de fábrica para criar uma classe que envolve uma nova operação TPUReplicateMetadata.

Parâmetros

escopo	escopo atual
numReplicas	Número de réplicas do cálculo
opções	carrega valores de atributos opcionais

Devoluções

• uma nova instância de TPUReplicateMetadata

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

Parâmetros

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

público estático TPUReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

público estático TPUReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica longo)

Parâmetros

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

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

público estático TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

Topologia pública estática TPUReplicateMetadata.Options (topologia String)

Parâmetros

topologia	TopologyProto indicando a topologia da fatia do pod TPU.

público estático TPUReplicateMetadata.Options tpuCompileOptionsProto (String tpuCompileOptionsProto)

public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)

público estático TPUReplicateMetadata.Options useTpu (booleano useTpu)

Parâmetros

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