Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

TPURoundRobin

kelas akhir publik TPURoundRobin

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

Output publik <Integer> deviceOrdinal ()