publiczna klasa statyczna ReplicateMetadata.Options
Opcjonalne atrybuty dla ReplicateMetadata
Metody publiczne
| Opcje replikacji metadanych | zezwolenieSoftPlacement (boolowski zezwolenieSoftPlacement) |
| Opcje replikacji metadanych | computationShape (List<Long> computationShape) |
| Opcje replikacji metadanych | DeviceAssignment (List<Long> DeviceAssignment) |
| Opcje replikacji metadanych | hostComputeCore (Lista<String> hostComputeCore) |
| Opcje replikacji metadanych | numCoresPerReplica (długie numCoresPerReplica) |
| Opcje replikacji metadanych | paddingMap (List<String> paddingMap) |
| Opcje replikacji metadanych | stepMarkerLocation (ciąg znaków stepMarkerLocation) |
| Opcje replikacji metadanych | topologia (topologia stringów) |
| Opcje replikacji metadanych | useSpmdForXlaPartitioning (boolean useSpmdForXlaPartitioning) |
| Opcje replikacji metadanych | useTpu (wartość logiczna useTpu) |
Metody dziedziczone
Metody publiczne
public ReplicateMetadata.Options computationShape (List<Long> computationShape)
Parametry
| obliczenieKształt | WYCOFANE. Zamiast tego użyj num_cores_per_replica. |
|---|
public ReplicateMetadata.Options DeviceAssignment (List<Long> DeviceAssignment)
Parametry
| Przypisanie urządzenia | Przypisanie urządzeń do obliczeń. |
|---|
public ReplicateMetadata.Options numCoresPerReplica (długie numCoresPerReplica)
Parametry
| liczba rdzeni na replikę | Liczba rdzeni na replikę. Używany do równoległości modelu. |
|---|
publiczna topologia ReplicateMetadata.Options (topologia String)
Parametry
| topologia | TopologyProto wskazujący topologię wycinka TPU. |
|---|
public ReplicateMetadata.Options useTpu (wartość logiczna useTpu)
Parametry
| użyjTpu | Czy umieścić obliczenia na TPU. |
|---|