public static class TPUReplicateMetadata.Options
Atributos opcionais para TPUReplicateMetadata
Métodos Públicos
TPUReplicateMetadata.Options | allowSoftPlacement (Boolean 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 | topologia ( topologia de string) |
TPUReplicateMetadata.Options | useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning) |
TPUReplicateMetadata.Options | useTpu (booleano useTpu) |
Métodos herdados
Métodos Públicos
public TPUReplicateMetadata.Options computationShape (List <Long> computationShape)
Parâmetros
computationShape | DESCONTINUADA. Em vez disso, use num_cores_per_replica. |
---|
public TPUReplicateMetadata.Options deviceAssignment (List <Long> deviceAssignment)
Parâmetros
deviceAssignment | A atribuição de dispositivos para a computação. |
---|
public TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
Parâmetros
numCoresPerReplica | Número de núcleos por réplica. Usado para paralelismo de modelo. |
---|
topologia public TPUReplicateMetadata.Options (topologia de string)
Parâmetros
topologia | TopologyProto indicando a topologia da fatia do pod de TPU. |
---|
public TPUReplicateMetadata.Options useTpu (Boolean useTpu)
Parâmetros
useTpu | Se deve colocar o cálculo na TPU. |
---|