TPURoundRobin

TPURoundRobin kelas akhir publik

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
membuat ( Lingkup ruang lingkup)
Metode pabrik untuk membuat kelas yang membungkus operasi TPURoundRobin baru.
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

Output publik <Bilangan Bulat> perangkatOrdinal ()