TPUReplicateMetadata.Options classe estática pública
Atributos opcionais para TPUReplicateMetadata
Métodos Públicos
| TPUReplicateMetadata.Options | allowSoftPlacement (booleano allowSoftPlacement) |
| TPUReplicateMetadata.Options | computationShape (List <longo> computationShape) |
| TPUReplicateMetadata.Options | deviceAssignment (List <longo> deviceAssignment) |
| TPUReplicateMetadata.Options | hostComputeCore (List <String> hostComputeCore) |
| TPUReplicateMetadata.Options | numCoresPerReplica (Long numCoresPerReplica) |
| TPUReplicateMetadata.Options | paddingMap (List <String> paddingMap) |
| TPUReplicateMetadata.Options | stepMarkerLocation (String stepMarkerLocation) |
| TPUReplicateMetadata.Options | topologia (topologia String) |
| TPUReplicateMetadata.Options | useSpmdForXlaPartitioning (booleano useSpmdForXlaPartitioning) |
| TPUReplicateMetadata.Options | useTpu (booleano useTpu) |
Métodos herdados
Métodos Públicos
pública TPUReplicateMetadata.Options computationShape (List <longo> computationShape)
Parâmetros
| computationShape | DESCONTINUADA. Em vez disso, use num_cores_per_replica. |
|---|
pública TPUReplicateMetadata.Options deviceAssignment (List <longo> deviceAssignment)
Parâmetros
| deviceAssignment | A atribuição de dispositivos para o cálculo. |
|---|
pública TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
Parâmetros
| numCoresPerReplica | Número de núcleos por réplica. Usado para paralelismo de modelo. |
|---|
pública TPUReplicateMetadata.Options topologia (topologia String)
Parâmetros
| topologia | TopologyProto indicando a topologia da fatia do pod de TPU. |
|---|
pública TPUReplicateMetadata.Options useTpu (Boolean useTpu)
Parâmetros
| useTpu | Se deve colocar o cálculo na TPU. |
|---|