Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

EnqueueTPUEmbeddingIntegerBatch

public class akhir EnqueueTPUEmbeddingIntegerBatch

Sebuah op yang mengantrekan daftar tensor batch input ke TPUEbedding.

Kelas Bersarang

kelas EnqueueTPUEbeddingIntegerBatch.Options Atribut opsional untuk EnqueueTPUEmbeddingIntegerBatch

Metode Publik

statis EnqueueTPUEmbeddingIntegerBatch
membuat ( Lingkup lingkup, Iterable < Operan <Integer >> batch, Operan <String> modeOverride, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingIntegerBatch baru.
statis EnqueueTPUEmbeddingIntegerBatch.Options
deviceOrdinal (Long deviceOrdinal)

Metode yang Diwarisi

Metode Publik

public static EnqueueTPUEmbeddingIntegerBatch membuat ( Scope lingkup, Iterable < Operan <Integer >> batch, Operan <String> modeOverride, Options ... pilihan)

Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingIntegerBatch baru.

Parameter
cakupan lingkup saat ini
kelompok Daftar tensor 1D, satu untuk setiap tabel penyematan, yang berisi indeks ke dalam tabel.
modeOverride Input string yang menggantikan mode yang ditentukan dalam TPUEmbbeddingConfiguration. Nilai yang didukung adalah {'unspecified', 'inference', 'training', 'backward_pass_only'}. Ketika diatur ke 'tidak ditentukan', mode yang diatur dalam TPUEmbeddingConfiguration digunakan, jika tidak, mode_override digunakan.
pilihan membawa nilai atribut opsional
Kembali
  • contoh baru dari EnqueueTPUEmbeddingIntegerBatch

public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)

Parameter
perangkatOrdinal Perangkat TPU yang akan digunakan. Harus >= 0 dan kurang dari jumlah inti TPU dalam tugas di mana node ditempatkan.