kelas akhir public InfeedEnqueuePrelinearizedBuffer
Sebuah op yang mengantrekan buffer yang telah di-linearisasikan ke dalam infeed TPU.
Kelas Bersarang
kelas | InfeedEnqueuePrelinearizedBuffer.Options | Atribut opsional untuk InfeedEnqueuePrelinearizedBuffer |
Metode Publik
statis InfeedEnqueuePrelinearizedBuffer | membuat ( Lingkup lingkup, Operan <?> input, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi InfeedEnqueuePrelinearizedBuffer baru. |
statis InfeedEnqueuePrelinearizedBuffer.Options | deviceOrdinal (Long deviceOrdinal) |
Metode yang Diwarisi
Metode Publik
public static InfeedEnqueuePrelinearizedBuffer membuat ( Scope lingkup, Operan <?> input, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi InfeedEnqueuePrelinearizedBuffer baru.
Parameter
cakupan | lingkup saat ini |
---|---|
memasukkan | Tensor varian yang mewakili keluaran linier. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari InfeedEnqueuePrelinearizedBuffer
public static InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Long deviceOrdinal)
Parameter
perangkatOrdinal | Perangkat TPU yang akan digunakan. Ini harus -1 ketika Op berjalan pada perangkat TPU dan = 0 ketika Op berjalan pada perangkat CPU. |
---|