kelas akhir publik InfeedEnqueuePrelinearizedBuffer
Sebuah operasi yang memasukkan buffer yang telah dipralinearisasi ke dalam pengumpan TPU.
Kelas Bersarang
kelas | InfeedEnqueuePrelinearizedBuffer.Options | Atribut opsional untuk InfeedEnqueuePrelinearizedBuffer |
Metode Publik
InfeedEnqueuePrelinearizedBuffer statis | |
InfeedEnqueuePrelinearizedBuffer.Options statis | deviceOrdinal (Perangkat panjangOrdinal) |
Metode Warisan
Metode Publik
public static InfeedEnqueuePrelinearizedBuffer membuat ( Lingkup lingkup, masukan Operan <?>, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi InfeedEnqueuePrelinearizedBuffer baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
memasukkan | Varian tensor yang mewakili keluaran yang dilinearisasi. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari InfeedEnqueuePrelinearizedBuffer
public static InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Perangkat PanjangOrdinal)
Parameter
perangkatOrdinal | Perangkat TPU yang akan digunakan. Nilainya harus -1 saat Operasi dijalankan pada perangkat TPU dan = 0 saat Operasi dijalankan pada perangkat CPU. |
---|