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