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

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	tpuCompileOptionsProto (Ciąg tpuCompileOptionsProto)
statyczne TPUReplicateMetadata.Options	useSpmdForXlaPartitioning (boolean useSpmdForXlaPartitioning)
statyczne TPUReplicateMetadata.Options	useTpu (wartość logiczna useTpu)

Metody dziedziczone

Z klasy org.tensorflow.op.PrimitiveOp

końcowa wartość logiczna	równa się (obiekt obiektu)
końcowy wew	hashCode ()
Operacja	op () Zwraca podstawową Operation
ostatni ciąg	doString ()

Z klasy java.lang.Object

wartość logiczna	równa się (Obiekt arg0)
ostatnia klasa<?>	pobierzKlasę ()
wew	hashCode ()
ostateczna pustka	powiadomić ()
ostateczna pustka	powiadom wszystkich ()
Strunowy	doString ()
ostateczna pustka	czekaj (długi arg0, int arg1)
ostateczna pustka	czekaj (długi arg0)
ostateczna pustka	Czekać ()

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 tpuCompileOptionsProto (String tpuCompileOptionsProto)

publiczny statyczny TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

publiczny statyczny TPUReplicateMetadata.Options useTpu (boolowski useTpu)

Parametry

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