InfeedEnqueuePrelinearizedBuffer

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
buat ( Lingkup lingkup, Operan <?> input, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi InfeedEnqueuePrelinearizedBuffer baru.
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.