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 subgráfico de cálculo `tpu.replicate()`.
Clases anidadas
| clase | TPUReplicateMetadata.Opciones | Atributos opcionales para TPUReplicateMetadata | |
Métodos públicos
| TPUReplicateMetadata.Opciones estáticas | allowSoftPlacement (booleano allowSoftPlacement) |
| TPUReplicateMetadata.Opciones estáticas | computationShape (List<Long> computationShape) |
| TPUReplicateMetadata estático | crear ( Ámbito de ámbito, Long numReplicas, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación TPUReplicateMetadata. |
| TPUReplicateMetadata.Opciones estáticas | asignación de dispositivo (lista <larga> asignación de dispositivo) |
| TPUReplicateMetadata.Opciones estáticas | hostComputeCore (Lista<String> hostComputeCore) |
| TPUReplicateMetadata.Opciones estáticas | numCoresPerReplica (Número largo de núcleos por réplica) |
| TPUReplicateMetadata.Opciones estáticas | paddingMap (Lista<String> paddingMap) |
| TPUReplicateMetadata.Opciones estáticas | stepMarkerLocation (Cadena stepMarkerLocation) |
| TPUReplicateMetadata.Opciones estáticas | topología (topología de cadena) |
| TPUReplicateMetadata.Opciones estáticas | tpuCompileOptionsProto (Cadena tpuCompileOptionsProto) |
| TPUReplicateMetadata.Opciones estáticas | useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning) |
| TPUReplicateMetadata.Opciones estáticas | useTpu (booleano useTpu) |
Métodos Heredados
Métodos públicos
public static TPUReplicateMetadata.Options computationShape (List<Long> computationShape)
Parámetros
| ComputaciónForma | OBSOLETO. Utilice num_cores_per_replica en su lugar. |
|---|
public static TPUReplicateMetadata create ( Scope scope, Long numReplicas, Options... options)
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
pública estática TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
Parámetros
| asignación de dispositivo | La asignación de dispositivos para el cómputo. |
|---|
public static TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
Parámetros
| numCoresPerReplica | Número de núcleos por réplica. Se utiliza para el paralelismo del modelo. |
|---|
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. |
|---|
público estático TPUReplicateMetadata.Options tpuCompileOptionsProto (String tpuCompileOptionsProto)
public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
public static TPUReplicateMetadata.Options useTpu (booleano useTpu)
Parámetros
| usoTpu | Si colocar el cálculo en la TPU. |
|---|