ReplicateMetadata

publiczna klasa końcowa ReplicateMetadata

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 replikacji metadanych Opcjonalne atrybuty dla ReplicateMetadata

Stałe

Strunowy OP_NAME Nazwa tej operacji znana silnikowi rdzenia TensorFlow

Metody publiczne

statyczne ReplicateMetadata.Options
zezwolenieSoftPlacement (boolowski zezwolenieSoftPlacement)
statyczne ReplicateMetadata.Options
computationShape (List<Long> computationShape)
statyczne replikacje metadanych
utwórz (zakres zakresu , Long numReplicas, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ReplicateMetadata.
statyczne ReplicateMetadata.Options
DeviceAssignment (List<Long> DeviceAssignment)
statyczne ReplicateMetadata.Options
hostComputeCore (Lista<String> hostComputeCore)
statyczne ReplicateMetadata.Options
numCoresPerReplica (długie numCoresPerReplica)
statyczne ReplicateMetadata.Options
paddingMap (List<String> paddingMap)
statyczne ReplicateMetadata.Options
stepMarkerLocation (ciąg znaków stepMarkerLocation)
statyczne ReplicateMetadata.Options
topologia (topologia stringów)
statyczne ReplicateMetadata.Options
useSpmdForXlaPartitioning (boolean useSpmdForXlaPartitioning)
statyczne ReplicateMetadata.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

public static ReplicateMetadata.Options zezwolenieSoftPlacement (boolowski zezwolenieSoftPlacement)

public static ReplicateMetadata.Options computationShape (List<Long> computationShape)

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

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

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

Parametry
zakres aktualny zakres
liczba replik Liczba replik obliczeń
opcje przenosi opcjonalne wartości atrybutów
Zwroty
  • nowe wystąpienie ReplicateMetadata

public static ReplicateMetadata.Options DeviceAssignment (List<Long> DeviceAssignment)

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

public static ReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

public static ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

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

public static ReplicateMetadata.Options paddingMap (List<String> paddingMap)

public static ReplicateMetadata.Options stepMarkerLocation (Ciąg stepMarkerLocation)

publiczna statyczna topologia ReplicateMetadata.Options (topologia String)

Parametry
topologia TopologyProto wskazujący topologię wycinka TPU.

public static ReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

public static ReplicateMetadata.Options useTpu (Boolean useTpu)

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