TPUReplicateMetadata

TPUReplicationMetadata 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 TPUReplikatMetadata.Opsi Atribut opsional untuk TPUReplicateMetadata

Metode Publik

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

Metode Warisan

Metode Publik

TPUReplicationMetadata.Options statis publik memungkinkanSoftPlacement (Boolean AllowSoftPlacement)

TPUReplicationMetadata.Options statis publik ComputationShape (Daftar<Panjang> ComputationShape)

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

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

Metode pabrik untuk membuat kelas yang membungkus operasi TPUReplicationMetadata baru.

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

TPUReplicationMetadata.Options statis publik penugasan perangkat (Daftar<Panjang> penugasan perangkat)

Parameter
penugasan perangkat Penugasan perangkat untuk komputasi.

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

TPUReplicationMetadata.Options statis publik numCoresPerReplica (NumCoresPerReplica panjang)

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

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

TPUReplicationMetadata.Options statis publik stepMarkerLocation (String stepMarkerLocation)

Topologi TPUReplicationMetadata.Options statis publik (Topologi string)

Parameter
topologi TopologyProto menunjukkan topologi potongan pod TPU.

TPUReplicationMetadata.Options statis publik tpuCompileOptionsProto (String tpuCompileOptionsProto)

TPUReplicationMetadata.Options statis publik useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

TPUReplicationMetadata statis publik. Opsi useTpu (Boolean useTpu)

Parameter
gunakanTpu Apakah akan menempatkan perhitungan pada TPU.