Menerima data deduplikasi (indeks dan bobot) dari inti embedding.
Data deduplikasi adalah Tensor dengan type=DT_VARIANT. Tensor itu sendiri adalah tupel bersarang XLA yang berisi elemen N (di mana 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.
Metode Publik
Keluaran <Objek> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
XlaRecvTPUEmbeddingDeduplicationData statis | |
Keluaran <?> | keluaran () |
Metode yang Diwarisi
Metode Publik
Output publik <Object> sebagai Output ()
Mengembalikan pegangan simbolis tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
Buat XlaRecvTPUEmbeddingDeduplicationData statis publik ( Cakupan lingkup, Konfigurasi string)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaRecvTPUEmbeddingDeduplicationData baru.
Parameter
cakupan | lingkup saat ini |
---|---|
konfigurasi | Proto konfigurasi TPUEbeddingConfiguration berseri. |
Kembali
- contoh baru dari XlaRecvTPUEmbeddingDeduplicationData