Op yang menjalankan program dengan pembaruan variabel opsional di tempat.
Ini (opsional) membaca variabel perangkat, memuat dan mengeksekusi program TPU pada perangkat TPU, dan kemudian (opsional) memperbarui variabel di tempat menggunakan output program, seperti yang ditentukan dalam atribut device_var_reads_indices (indeks input program dari variabel pembacaan langsung) dan device_var_updates_indices (indeks keluaran program digunakan untuk memperbarui variabel, -1 berarti tidak ada pembaruan/hanya-baca). Output program yang dikonsumsi oleh variabel-variabel ini tidak akan muncul di output operasi. Untuk penggunaan internal kompiler TPU terdistribusi.
Metode Publik
TPUExecuteAndUpdateVariables statis | |
Iterator< Operan <Objek>> | pengulangan () |
Daftar< Keluaran <?>> | hasil () |
Metode Warisan
Metode Publik
public static TPUExecuteAndUpdateVariables dibuat ( Cakupan cakupan , Iterable< Operand <?>> args, kunci Operand <String>, Daftar<Kelas<?>> Hasil, Daftar<Panjang> perangkatVarReadsIndices, Daftar<Panjang> perangkatVarUpdatesIndices)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUExecuteAndUpdateVariables baru.
Parameter
cakupan | ruang lingkup saat ini |
---|
Kembali
- contoh baru dari TPUExecuteAndUpdateVariables