XlaRecvTPUEmbeddingActivations

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

אופ שמקבל הפעלות הטמעה ב-TPU.

מערכת ה-TPU מבצעת את חיפושי הטבעה והצטברויות. התוצאות של צבירה אלו גלויות לגרף Tensorflow כפלטים של XlaRecvTPUEmbeddingActivations Op. פעולה זו מחזירה רשימה המכילה Tensor אחד של הפעלות לכל טבלה שצוינה במודל.

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

סטטי XlaRecvTPUEmbeddingActivations
create ( scope scope, Operand <?> deduplicationData, Long numTables, String config)
שיטת מפעל ליצירת מחלקה העוטפת פעולת XlaRecvTPUEmbeddingActivations חדשה.
איטרטור< Oprand <Float>>
רשימה< פלט <צף>>
פלטים ()
רשימת Tensor של הפעלות הטמעה המכילה Tensor אחד לכל טבלת הטמעה במודל.

שיטות בירושה

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

ציבורי סטטי XlaRecvTPUEmbeddingActivations create ( scope scope, Operand <?> deduplicationData, Long numTables, String config)

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

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

איטרטור ציבורי< Operand <Float>> איטרטור ()

רשימה ציבורית< פלט <צף>> פלטי ()

רשימת Tensor של הפעלות הטמעה המכילה Tensor אחד לכל טבלת הטמעה במודל.