Penyeimbangan beban round-robin pada inti TPU.
Operasi penyeimbangan beban yang dilakukan secara round-robin di antara inti TPU.
Operasi ini dilakukan secara round-robin antara bilangan bulat di [0, NumTPUCoresVisiblePerHost]. Hal ini berguna untuk berinteraksi dengan operasi TensorFlow yang menggunakan inti TPU sebagai masukan untuk menjalankan komputasi, seperti `TPUPartitionedCall`.
device_ordinal: Bilangan bulat dalam [0, NumTPUCoresVisiblePerHost].
Metode Publik
Keluaran <Bilangan Bulat> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
TPURoundRobin statis | |
Keluaran <Bilangan Bulat> |
Metode Warisan
Metode Publik
Keluaran publik <Bilangan Bulat> sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
pembuatan TPURoundRobin statis publik ( Lingkup lingkup)
Metode pabrik untuk membuat kelas yang membungkus operasi TPURoundRobin baru.
Parameter
cakupan | ruang lingkup saat ini |
---|
Kembali
- contoh baru TPURoundRobin