TPUReplicateMetadata

publiczna klasa końcowa TPUReplicateMetadata

Metadane wskazujące, w jaki sposób należy replikować obliczenia TPU.

Ta operacja przechowuje metadane wspólne dla operacji podgrafu obliczeniowego `tpu.replicate()`.

Klasy zagnieżdżone

klasa Opcje TPUReplicateMetadata Opcjonalne atrybuty dla TPUReplicateMetadata

Stałe

Strunowy OP_NAME Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Metody publiczne

statyczne TPUReplicateMetadata.Options
zezwolenieSoftPlacement (boolowski zezwolenieSoftPlacement)
statyczne TPUReplicateMetadata.Options
computationShape (List<Long> computationShape)
statyczne TPUReplicateMetadata
utwórz (zakres zakresu , Long numReplicas, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUReplicateMetadata.
statyczne TPUReplicateMetadata.Options
DeviceAssignment (List<Long> DeviceAssignment)
statyczne TPUReplicateMetadata.Options
hostComputeCore (Lista<String> hostComputeCore)
statyczne TPUReplicateMetadata.Options
numCoresPerReplica (długie numCoresPerReplica)
statyczne TPUReplicateMetadata.Options
paddingMap (List<String> paddingMap)
statyczne TPUReplicateMetadata.Options
stepMarkerLocation (ciąg znaków stepMarkerLocation)
statyczne TPUReplicateMetadata.Options
topologia (topologia stringów)
statyczne TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (boolean useSpmdForXlaPartitioning)
statyczne TPUReplicateMetadata.Options
useTpu (wartość logiczna useTpu)

Metody dziedziczone

Stałe

publiczny statyczny końcowy ciąg znaków OP_NAME

Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Wartość stała: „TPUReplicateMetadata”

Metody publiczne

publiczny statyczny TPUReplicateMetadata.Options zezwolenieSoftPlacement (boolowski zezwolenieSoftPlacement)

publiczny statyczny TPUReplicateMetadata.Options computationShape (List<Long> computationShape)

Parametry
obliczenieKształt WYCOFANE. Zamiast tego użyj num_cores_per_replica.

public static TPUReplicateMetadata create ( Zakres zakresu, Long numReplicas, Opcje... opcje)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUReplicateMetadata.

Parametry
zakres aktualny zakres
liczba replik Liczba replik obliczeń
opcje przenosi opcjonalne wartości atrybutów
Zwroty
  • nowa instancja TPUReplicateMetadata

publiczne statyczne TPUReplicateMetadata.Options DeviceAssignment (List<Long> DeviceAssignment)

Parametry
Przypisanie urządzenia Przypisanie urządzeń do obliczeń.

publiczny statyczny TPUReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

public static TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

Parametry
liczba rdzeni na replikę Liczba rdzeni na replikę. Używany do równoległości modelu.

publiczna statyczna TPUReplicateMetadata.Options paddingMap (List<String> paddingMap)

public static TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

publiczna statyczna topologia TPUReplicateMetadata.Options (topologia String)

Parametry
topologia TopologyProto wskazujący topologię wycinka TPU.

publiczny statyczny TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

publiczny statyczny TPUReplicateMetadata.Options useTpu (boolowski useTpu)

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