clase estática pública TPUReplicateMetadata.Options
Atributos opcionales para TPUReplicateMetadata
Métodos públicos
TPUReplicateMetadata.Options | allowSoftPlacement (booleano allowSoftPlacement) |
TPUReplicateMetadata.Options | computationShape (List <Long> computationShape) |
TPUReplicateMetadata.Options | deviceAssignment (List <Long> deviceAssignment) |
TPUReplicateMetadata.Options | hostComputeCore (Lista <String> hostComputeCore) |
TPUReplicateMetadata.Options | numCoresPerReplica (Long numCoresPerReplica) |
TPUReplicateMetadata.Options | paddingMap (List <String> paddingMap) |
TPUReplicateMetadata.Options | stepMarkerLocation (String stepMarkerLocation) |
TPUReplicateMetadata.Options | topología ( topología de cadenas) |
TPUReplicateMetadata.Options | useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning) |
TPUReplicateMetadata.Options | useTpu (booleano useTpu) |
Métodos heredados
Métodos públicos
public TPUReplicateMetadata.Options computationShape (List <Long> computationShape)
Parámetros
computationShape | OBSOLETO. Utilice num_cores_per_replica en su lugar. |
---|
public TPUReplicateMetadata.Options deviceAssignment (List <Long> deviceAssignment)
Parámetros
deviceAssignment | La asignación de dispositivos para el cálculo. |
---|
public TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
Parámetros
numCoresPerReplica | Número de núcleos por réplica. Se utiliza para el paralelismo de modelos. |
---|
topología pública TPUReplicateMetadata.Options (topología de cadenas)
Parámetros
topología | TopologyProto que indica la topología del segmento del pod de TPU. |
---|
public TPUReplicateMetadata.Options useTpu (Boolean useTpu)
Parámetros
useTpu | Ya sea para colocar el cálculo en la TPU. |
---|