clase final pública TPUReplicateMetadata
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 | Opciones de TPUReplicateMetadata | Atributos opcionales para TPUReplicateMetadata | |
Métodos públicos
| TPUReplicateMetadata estático.Opciones | permitirSoftPlacement (permiso booleanoSoftPlacement) |
| TPUReplicateMetadata estático.Opciones | forma de cálculo (Lista<Long> forma de cálculo) |
| TPUReplicateMetadata estático | crear (alcance alcance , Long numReplicas, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación TPUReplicateMetadata. |
| TPUReplicateMetadata estático.Opciones | Asignación de dispositivo (Lista<Long> asignación de dispositivo) |
| TPUReplicateMetadata estático.Opciones | hostComputeCore (Lista<Cadena> hostComputeCore) |
| TPUReplicateMetadata estático.Opciones | numCoresPerReplica (numCoresPerReplica largo) |
| TPUReplicateMetadata estático.Opciones | paddingMap (Lista<String> paddingMap) |
| TPUReplicateMetadata estático.Opciones | stepMarkerLocation (cadena stepMarkerLocation) |
| TPUReplicateMetadata estático.Opciones | topología (topología de cadena) |
| TPUReplicateMetadata estático.Opciones | tpuCompileOptionsProto (cadena tpuCompileOptionsProto) |
| TPUReplicateMetadata estático.Opciones | useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning) |
| TPUReplicateMetadata estático.Opciones | useTpu (uso booleanoTpu) |
Métodos heredados
Métodos públicos
TPUReplicateMetadata.Options estático público computationShape (List<Long> computationShape)
Parámetros
| cálculoForma | OBSOLETO. Utilice num_cores_per_replica en su lugar. |
|---|
Creación pública estática de TPUReplicateMetadata ( alcance , número largo de réplicas, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación TPUReplicateMetadata.
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 TPUReplicateMetadata
TPUReplicateMetadata.Options estática pública asignación de dispositivo (Lista <Long> asignación de dispositivo)
Parámetros
| dispositivoAsignación | La asignación de dispositivos para el cómputo. |
|---|
TPUReplicateMetadata.Options estático público 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 TPUReplicateMetadata.Options (topología de cadena)
Parámetros
| topología | TopologyProto que indica la topología del segmento del pod de TPU. |
|---|
TPUReplicateMetadata.Options estático público tpuCompileOptionsProto (String tpuCompileOptionsProto)
TPUReplicateMetadata.Options estático público useSpmdForXlaPartitioning (uso booleanoSpmdForXlaPartitioning)
TPUReplicateMetadata público estático. Opciones useTpu (uso booleano Tpu)
Parámetros
| utilizarTpu | Ya sea para colocar el cálculo en la TPU. |
|---|