clase final pública ReplicateMetadata
Metadatos que indican cómo se debe replicar el cálculo de TPU.
Esta operación contiene los metadatos comunes a las operaciones de un subgrafo de cálculo `tpu.replicate()`.
Clases anidadas
| clase | ReplicarMetadatos.Opciones | Atributos opcionales para ReplicateMetadata | |
Constantes
| Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
| Opciones estáticas de ReplicateMetadata. | permitirSoftPlacement (permiso booleanoSoftPlacement) |
| Opciones estáticas de ReplicateMetadata. | forma de cálculo (Lista<Long> forma de cálculo) |
| ReplicarMetadatos estáticos | crear (alcance alcance , Long numReplicas, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación ReplicateMetadata. |
| Opciones estáticas de ReplicateMetadata. | Asignación de dispositivo (Lista<Long> asignación de dispositivo) |
| Opciones estáticas de ReplicateMetadata. | hostComputeCore (Lista<Cadena> hostComputeCore) |
| Opciones estáticas de ReplicateMetadata. | numCoresPerReplica (numCoresPerReplica largo) |
| Opciones estáticas de ReplicateMetadata. | paddingMap (Lista<String> paddingMap) |
| Opciones estáticas de ReplicateMetadata. | stepMarkerLocation (cadena stepMarkerLocation) |
| Opciones estáticas de ReplicateMetadata. | topología (topología de cadena) |
| Opciones estáticas de ReplicateMetadata. | useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning) |
| Opciones estáticas de ReplicateMetadata. | useTpu (uso booleanoTpu) |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Valor constante: "TPUReplicateMetadata"
Métodos públicos
pública estática ReplicateMetadata.Options computationShape (List<Long> computationShape)
Parámetros
| cálculoForma | OBSOLETO. Utilice num_cores_per_replica en su lugar. |
|---|
creación pública estática de ReplicateMetadata ( alcance , número largo de réplicas, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación ReplicateMetadata.
Parámetros
| alcance | alcance actual |
|---|---|
| número de réplicas | Número de réplicas del cálculo. |
| opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de ReplicateMetadata
public static ReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
Parámetros
| dispositivoAsignación | La asignación de dispositivos para el cómputo. |
|---|
público estático ReplicateMetadata.Options numCoresPerReplica (NumCoresPerReplica largo)
Parámetros
| número de núcleos por réplica | Número de núcleos por réplica. Se utiliza para el paralelismo de modelos. |
|---|
Topología pública estática ReplicateMetadata.Options (topología de cadena)
Parámetros
| topología | TopologyProto que indica la topología del segmento del pod de TPU. |
|---|
público estático ReplicateMetadata.Options useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)
público estático ReplicateMetadata.Options useTpu (useTpu booleano)
Parámetros
| utilizarTpu | Ya sea para colocar el cálculo en la TPU. |
|---|