Menerima data deduplikasi (indeks dan bobot) dari inti penyematan.
Data deduplikasinya adalah Tensor dengan type=DT_VARIANT. Tensor itu sendiri adalah tupel bersarang XLA yang berisi N elemen (di mana N adalah rasio jumlah penyematan inti tensor per chip TPU). Setiap elemen tupel bersarang adalah tupel dengan tensor peringkat 1. Setiap tensor berisi indeks (DT_UINT32) untuk menyematkan pencarian di TensorCore atau bobot (DT_FLOAT) untuk diterapkan ke output dari operasi pencarian penyematan.
Metode Publik
Keluaran <Objek> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
XlaRecvTPUEmbeddingDeduplicationData statis | |
Keluaran <?> | keluaran () |
Metode Warisan
Metode Publik
Output publik <Objek> asOutput ()
Mengembalikan pegangan simbolik tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
pembuatan XlaRecvTPUEmbeddingDeduplicationData statis publik ( Lingkup cakupan, Konfigurasi string)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaRecvTPUEmbeddingDeduplicationData baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
konfigurasi | Proto TPUEmbeddingConfiguration berseri. |
Kembali
- contoh baru dari XlaRecvTPUEmbeddingDeduplicationData