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 | |
Konstanta
| Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
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 | useSpmdForXlaPartitioning (penggunaan BooleanSpmdForXlaPartitioning) |
| TPUReplicationMetadata.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
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. |
|---|