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 | |
Constantes
| Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
| TPUReplicateMetadata.Options estático | permitirSoftPlacement (booleano permitirSoftPlacement) |
| TPUReplicateMetadata.Options estático | computationShape (List<Long> computationShape) |
| TPUReplicateMetadata estático | |
| 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 | useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning) |
| TPUReplicateMetadata.Options estático | useTpu (booleano useTpu) |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Valor constante: "TPUReplicateMetadata"
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 numCoresPerReplica (numCoresPerReplica longo)
Parâmetros
| numCoresPerReplica | Número de núcleos por réplica. Usado para paralelismo de modelo. |
|---|
Topologia pública estática TPUReplicateMetadata.Options (topologia String)
Parâmetros
| topologia | TopologyProto indicando a topologia da fatia do pod TPU. |
|---|
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. |
|---|