Operasi yang menerima aktivasi penyematan pada TPU.
Sistem TPU melakukan pencarian dan agregasi embedding. Hasil dari agregasi ini terlihat oleh Grafik Tensorflow sebagai output dari XlaRecvTPUEmbeddingActivations Op. Operasi ini mengembalikan daftar yang berisi satu Tensor aktivasi per tabel yang ditentukan dalam model.
Metode Publik
XlaRecvTPUEmbeddingActivations statis | |
Iterator< Operan <Float>> | pengulangan () |
Daftar< Keluaran <Float>> | keluaran () TensorList aktivasi penyematan yang berisi satu Tensor per tabel penyematan dalam model. |
Metode yang Diwarisi
Metode Publik
Buat XlaRecvTPUEmbeddingActivations statis publik ( Cakupan lingkup, Operand <?> deduplicationData, numTables panjang, Konfigurasi string)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaRecvTPUEmbbeddingActivations baru.
Parameter
cakupan | lingkup saat ini |
---|---|
data deduplikasi | Tensor dengan type=DT_VARIANT yang berisi data deduplikasi. Tensor adalah tupel bersarang XLA yang berisi elemen N (dengan N adalah rasio jumlah penyematan ke inti tensor per chip TPU). Setiap elemen tupel bersarang adalah tupel tensor peringkat 1. Setiap tensor berisi indeks (DT_UINT32) untuk menyematkan pencarian pada TensorCore atau bobot (DT_FLOAT) untuk diterapkan ke output dari operasi pencarian penyematan. |
angkaTabel | Jumlah tensor aktivasi keluaran. Jika deskriptor fitur ada dalam konfigurasi embedding tpu, itu sama dengan jumlah fitur jika tidak sama dengan jumlah tabel embedding dalam model. |
konfigurasi | Proto konfigurasi TPUEbeddingConfiguration berseri. |
Kembali
- contoh baru dari XlaRecvTPUEmbeddingActivations
Daftar publik< Output <Float>> output ()
TensorList aktivasi penyematan yang berisi satu Tensor per tabel penyematan dalam model.