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 | |
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 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 numCoresPerReplica (NumCoresPerReplica panjang)
Parameter
numCoresPerReplika | Jumlah inti per replika. Digunakan untuk paralelisme model. |
---|
Topologi TPUReplicationMetadata.Options statis publik (Topologi string)
Parameter
topologi | TopologyProto menunjukkan topologi potongan pod TPU. |
---|
TPUReplicationMetadata.Options statis publik useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
TPUReplicationMetadata statis publik. Opsi useTpu (Boolean useTpu)
Parameter
gunakanTpu | Apakah akan menempatkan perhitungan pada TPU. |
---|