Clase public final TPUReplicateMetadata
Metadatos que indican cómo se debe replicar el cálculo de la TPU.
Esta operación contiene los metadatos comunes a las operaciones de un subgrafo de cálculo `tpu.replicate ()`.
Clases anidadas
| clase | TPUReplicateMetadata.Options | Los atributos opcionales para TPUReplicateMetadata | |
Métodos públicos
| estáticas TPUReplicateMetadata.Options | allowSoftPlacement (Boolean allowSoftPlacement) |
| estáticas TPUReplicateMetadata.Options | computationShape (List <Larga> computationShape) |
| estática TPUReplicateMetadata | crear ( Alcance alcance, largas numReplicas, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación TPUReplicateMetadata. |
| estáticas TPUReplicateMetadata.Options | deviceAssignment (List <Larga> deviceAssignment) |
| estáticas TPUReplicateMetadata.Options | hostComputeCore (List <String> hostComputeCore) |
| estáticas TPUReplicateMetadata.Options | numCoresPerReplica (Long numCoresPerReplica) |
| estáticas TPUReplicateMetadata.Options | paddingMap (List <String> paddingMap) |
| estáticas TPUReplicateMetadata.Options | stepMarkerLocation (String stepMarkerLocation) |
| estáticas TPUReplicateMetadata.Options | topología (topología String) |
| estáticas TPUReplicateMetadata.Options | useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning) |
| estáticas TPUReplicateMetadata.Options | useTpu (Boolean useTpu) |
Métodos heredados
Métodos públicos
public static TPUReplicateMetadata.Options computationShape (List <Larga> computationShape)
Parámetros
| computationShape | OBSOLETO. Utilice num_cores_per_replica en su lugar. |
|---|
public static TPUReplicateMetadata crear ( Alcance alcance, Long numReplicas, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación TPUReplicateMetadata.
Parámetros
| alcance | alcance actual |
|---|---|
| numReplicas | Número de réplicas del cálculo |
| opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de TPUReplicateMetadata
public static TPUReplicateMetadata.Options deviceAssignment (List <Larga> deviceAssignment)
Parámetros
| deviceAssignment | La asignación de dispositivos para el cálculo. |
|---|
public static TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
Parámetros
| numCoresPerReplica | Número de núcleos por réplica. Se utiliza para el paralelismo de modelos. |
|---|
estáticas pública TPUReplicateMetadata.Options topología (topología de cadena)
Parámetros
| topología | TopologyProto que indica la topología del segmento del pod de TPU. |
|---|
public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning)
public static TPUReplicateMetadata.Options useTpu (booleano useTpu)
Parámetros
| useTpu | Ya sea para colocar el cálculo en la TPU. |
|---|