Penyeimbangan beban round-robin pada inti TPU.
Operasi penyeimbangan beban yang melakukan round-robin di antara inti TPU.
Ini round-robin op antara bilangan bulat di [0, NumTPUCoresVisiblePerHost]. Ini berguna untuk berinteraksi dengan operasi TensorFlow yang mengambil inti TPU sebagai input untuk menjalankan komputasi, seperti `TPUPartitionedCall`.
device_ordinal: Sebuah bilangan bulat di [0, NumTPUCoresVisiblePerHost].
Metode Publik
Keluaran <Bilangan Bulat> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
TPURoundRobin statis | |
Keluaran <Bilangan Bulat> |
Metode yang Diwarisi
Metode Publik
Output publik <Integer> sebagaiOutput ()
Mengembalikan pegangan simbolis tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
pembuatan TPURoundRobin statis publik ( Cakupan lingkup)
Metode pabrik untuk membuat kelas yang membungkus operasi TPURoundRobin baru.
Parameter
cakupan | lingkup saat ini |
---|
Kembali
- contoh baru TPURoundRobin