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