ReplicateMetadata

ReplikasiMetadata kelas akhir publik

Metadata yang menunjukkan bagaimana komputasi TPU harus direplikasi.

Operasi ini menyimpan metadata yang umum untuk operasi subgraf komputasi `tpu.replication()`.

Kelas Bersarang

kelas ReplikasiMetadata.Options Atribut opsional untuk ReplicateMetadata

Konstanta

Rangkaian OP_NAME Nama operasi ini dikenal dengan mesin inti TensorFlow

Metode Publik

ReplikasiMetadata.Options statis
izinkanSoftPlacement (Boolean izinkanSoftPlacement)
ReplikasiMetadata.Options statis
ComputationShape (Daftar<Panjang> ComputationShape)
Metadata Replikasi statis
buat ( Lingkup cakupan, Jumlah Replika panjang, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi ReplicationMetadata baru.
ReplikasiMetadata.Options statis
penugasan perangkat (Daftar<Panjang> penugasan perangkat)
ReplikasiMetadata.Options statis
hostComputeCore (Daftar<String> hostComputeCore)
ReplikasiMetadata.Options statis
numCoresPerReplica (NumCoresPerReplika panjang)
ReplikasiMetadata.Options statis
paddingMap (Daftar<String> paddingMap)
ReplikasiMetadata.Options statis
stepMarkerLocation (String stepMarkerLocation)
ReplikasiMetadata.Options statis
topologi (Topologi string)
ReplikasiMetadata.Options statis
useSpmdForXlaPartitioning (penggunaan BooleanSpmdForXlaPartitioning)
ReplikasiMetadata.Options statis
useTpu (Boolean useTpu)

Metode Warisan

Konstanta

String akhir statis publik OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Nilai Konstan: "TPUReplicationMetadata"

Metode Publik

ReplikasiMetadata statis publik.OptionsallowSoftPlacement (BooleanallowSoftPlacement)

public static ReplicationMetadata.Options ComputationShape (Daftar<Panjang> ComputationShape)

Parameter
bentuk komputasi TIDAK DIGUNAKAN lagi. Gunakan num_cores_per_replica sebagai gantinya.

pembuatan ReplikasiMetadata statis publik ( Lingkup cakupan, JumlahReplika panjang, Opsi... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi ReplicationMetadata baru.

Parameter
cakupan ruang lingkup saat ini
nomorReplika Jumlah replika komputasi
pilihan membawa nilai atribut opsional
Kembali
  • contoh baru dari ReplikasiMetadata

ReplikasiMetadata.Opsi statis publik Penugasan perangkat (Daftar<Panjang> Penugasan perangkat)

Parameter
penugasan perangkat Penugasan perangkat untuk komputasi.

ReplikasiMetadata.Options statis publik hostComputeCore (Daftar<String> hostComputeCore)

ReplikasiMetadata.Options statis publik numCoresPerReplica (NumCoresPerReplica panjang)

Parameter
numCoresPerReplika Jumlah inti per replika. Digunakan untuk paralelisme model.

ReplikasiMetadata.Options paddingMap statis publik (Daftar<String> paddingMap)

ReplikasiMetadata statis publik.Opsi stepMarkerLocation (String stepMarkerLocation)

topologi ReplikasiMetadata.Options statis publik (Topologi string)

Parameter
topologi TopologyProto menunjukkan topologi potongan pod TPU.

ReplikasiMetadata statis publik.Opsi useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

Replikasi Metadata statis publik. Opsi useTpu (Boolean useTpu)

Parameter
gunakanTpu Apakah akan menempatkan perhitungan pada TPU.