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
| 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 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. |
|---|
público ReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica longo)
Parâmetros
| numCoresPerReplica | Número de núcleos por réplica. Usado para paralelismo de modelo. |
|---|
Topologia pública ReplicateMetadata.Options (topologia String)
Parâmetros
| topologia | TopologyProto indicando a topologia da fatia do pod TPU. |
|---|
public ReplicateMetadata.Options useTpu (booleano useTpu)
Parâmetros
| usarTpu | Se o cálculo deve ser colocado na TPU. |
|---|