clase final pública 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 | Atributos opcionales para TPUReplicateMetadata |
Métodos públicos
TPUReplicateMetadata.Options estático | allowSoftPlacement (booleano allowSoftPlacement) |
TPUReplicateMetadata.Options estático | computationShape (List <Long> computationShape) |
TPUReplicateMetadata estático | crear ( alcance del alcance, numReplicas largas, Opciones ... opciones) Método de fábrica para crear una clase que envuelva una nueva operación TPUReplicateMetadata. |
TPUReplicateMetadata.Options estático | deviceAssignment (List <Long> deviceAssignment) |
TPUReplicateMetadata.Options estático | hostComputeCore (Lista <String> hostComputeCore) |
TPUReplicateMetadata.Options estático | numCoresPerReplica (Long numCoresPerReplica) |
TPUReplicateMetadata.Options estático | paddingMap (List <String> paddingMap) |
TPUReplicateMetadata.Options estático | stepMarkerLocation (String stepMarkerLocation) |
TPUReplicateMetadata.Options estático | topología ( topología de cadenas) |
TPUReplicateMetadata.Options estático | useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning) |
TPUReplicateMetadata.Options estático | useTpu (booleano useTpu) |
Métodos heredados
Métodos públicos
público estático TPUReplicateMetadata.Options computationShape (List <Long> computationShape)
Parámetros
computationShape | OBSOLETO. Utilice num_cores_per_replica en su lugar. |
---|
Public static TPUReplicateMetadata create ( Alcance del alcance, Long numReplicas, Options ... options)
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
TPUReplicateMetadata.Options estático público deviceAssignment (List <Long> deviceAssignment)
Parámetros
deviceAssignment | La asignación de dispositivos para el cálculo. |
---|
TPUReplicateMetadata.Options estática pública numCoresPerReplica (Long numCoresPerReplica)
Parámetros
numCoresPerReplica | Número de núcleos por réplica. Se utiliza para el paralelismo de modelos. |
---|
TPUReplicateMetadata.Options estática pública topología (topología de cadenas)
Parámetros
topología | TopologyProto que indica la topología del segmento del pod de TPU. |
---|
TPUReplicateMetadata.Options estática pública useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
TPUReplicateMetadata.Options estática pública useTpu (booleano useTpu)
Parámetros
useTpu | Ya sea para colocar el cálculo en la TPU. |
---|