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