public final class TPUReplicateMetadata
Metadados que indicam como a computação da TPU deve ser replicada.
Esta operação manté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
| static TPUReplicateMetadata.Options | allowSoftPlacement (Boolean allowSoftPlacement) |
| static TPUReplicateMetadata.Options | computationShape (List <Long> computationShape) |
| TPUReplicateMetadata estático | criar ( escopo do escopo, numReplicas longas, opções ... opções) Método de fábrica para criar uma classe envolvendo uma nova operação TPUReplicateMetadata. |
| static TPUReplicateMetadata.Options | deviceAssignment (List <Long> deviceAssignment) |
| static TPUReplicateMetadata.Options | hostComputeCore (Lista <String> hostComputeCore) |
| static TPUReplicateMetadata.Options | numCoresPerReplica (Long numCoresPerReplica) |
| static TPUReplicateMetadata.Options | paddingMap (List <String> paddingMap) |
| static TPUReplicateMetadata.Options | stepMarkerLocation (String stepMarkerLocation) |
| static TPUReplicateMetadata.Options | topologia ( topologia de string) |
| static TPUReplicateMetadata.Options | useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning) |
| static TPUReplicateMetadata.Options | useTpu (booleano useTpu) |
Métodos herdados
Métodos Públicos
public static TPUReplicateMetadata.Options computationShape (List <Long> computationShape)
Parâmetros
| computationShape | DESCONTINUADA. Em vez disso, use num_cores_per_replica. |
|---|
public static TPUReplicateMetadata create ( Scope scope, Long numReplicas, Options ... options)
Método de fábrica para criar uma classe envolvendo uma nova operação TPUReplicateMetadata.
Parâmetros
| alcance | 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
| deviceAssignment | A atribuição de dispositivos para a computação. |
|---|
public static TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
Parâmetros
| numCoresPerReplica | Número de núcleos por réplica. Usado para paralelismo de modelo. |
|---|
public static TPUReplicateMetadata.Options topology (String topology)
Parâmetros
| topologia | TopologyProto indicando a topologia da fatia do pod de TPU. |
|---|
public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
public static TPUReplicateMetadata.Options useTpu (Boolean useTpu)
Parâmetros
| useTpu | Se deve colocar o cálculo na TPU. |
|---|