XlaRecvTPUEmbeddingDeduplicationData

מחלקה סופית ציבורית XlaRecvTPUEmbeddingDeduplicationData

מקבל נתוני מניעת כפילות (מדדים ומשקלים) מליבת ההטמעה.

נתוני מניעת הכפילויות הם Tensor עם type=DT_VARIANT. הטנזור עצמו הוא tuple מקוננת XLA המכיל N אלמנטים (כאשר N הוא היחס בין מספר ליבות הטבעה לטנזור לכל שבב TPU). כל אלמנט של tuple מקוננת הוא tuple בדרגה 1 טנסור. כל טנסור מכיל מדדים (DT_UINT32) להטמעת חיפוש ב-TensorCore או משקלים (DT_FLOAT) שיחולו על הפלט של פעולת בדיקת ההטמעה.

שיטות ציבוריות

פלט <Object>
asOutput ()
מחזירה את הידית הסמלית של טנזור.
סטטי XlaRecvTPUEmbeddingDeduplicationData
ליצור (היקף היקף , תצורת מחרוזת)
שיטת מפעל ליצירת מחלקה העוטפת פעולה חדשה XlaRecvTPUEmbeddingDeduplicationData.
פלט <?>
פלט ()

שיטות בירושה

שיטות ציבוריות

פלט ציבורי <Object> asOutput ()

מחזירה את הידית הסמלית של טנזור.

כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.

סטטי ציבורי XlaRecvTPUEmbeddingDeduplicationData create (היקף היקף , תצורת מחרוזת)

שיטת מפעל ליצירת מחלקה העוטפת פעולה חדשה XlaRecvTPUEmbeddingDeduplicationData.

פרמטרים
תְחוּם ההיקף הנוכחי
תצורה פרוטו TPUEmbeddingConfiguration מסודר.
החזרות
  • מופע חדש של XlaRecvTPUEmbeddingDeduplicationData

פלט ציבורי <?> פלט ()