TPUReplicateMetadata.Options

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
useSpmdForXlaPartitioning (boolean useSpmdForXlaPartitioning)
Opcje TPUReplicateMetadata
useTpu (wartość logiczna useTpu)

Metody dziedziczone

Metody publiczne

public TPUReplicateMetadata.Options zezwolenieSoftPlacement (boolowski zezwolenieSoftPlacement)

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 hostComputeCore (List<String> hostComputeCore)

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 TPUReplicateMetadata.Options paddingMap (List<String> paddingMap)

public TPUReplicateMetadata.Options stepMarkerLocation (ciąg znaków stepMarkerLocation)

publiczna topologia TPUReplicateMetadata.Options (topologia String)

Parametry
topologia TopologyProto wskazujący topologię wycinka TPU.

public TPUReplicateMetadata.Options useSpmdForXlaPartitioning (boolean useSpmdForXlaPartitioning)

public TPUReplicateMetadata.Options useTpu (wartość logiczna useTpu)

Parametry
użyjTpu Czy umieścić obliczenia na TPU.