Selektor inti TPU Op.
Op ini menghasilkan satu set inti TPU (untuk pemanasan) atau satu inti TPU (untuk inferensi reguler) untuk menjalankan program TPU. Outputnya dikonsumsi oleh TPUPartitionedCall.
Metode Publik
Output <Integer> | asOutput () Mengembalikan pegangan simbolis tensor. |
statis TPUOrdinalSelector | |
Output <Integer> | deviceOrdinals () Sebuah vektor 1 atau lebih inti TPU. |
Metode yang Diwarisi
Metode Publik
publik Keluaran <Integer> asOutput ()
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.
public static TPUOrdinalSelector membuat ( Scope lingkup)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUOrdinalSelector baru.
Parameter
cakupan | lingkup saat ini |
---|
Kembali
- contoh baru dari TPUOrdinalSelector