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 | |
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 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 numCoresPerReplica (Long numCoresPerReplica)
Parametry
liczba rdzeni na replikę | Liczba rdzeni na replikę. Używany do równoległości modelu. |
---|
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. |
---|