אופ שמקבל הפעלות הטמעה ב-TPU.
מערכת ה-TPU מבצעת את חיפושי ההטמעה והצבירה שצוינו על ידי הארגומנטים ל-TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor) Batch. התוצאות של צבירה אלו גלויות לגרף Tensorflow כפלטים של הפעלה של RecvTPUEembeddingActivations. פעולה זו מחזירה רשימה המכילה Tensor אחד של הפעלות לכל טבלה שצוינה במודל. יכולה להיות הפעלה אחת של RecvTPUEmbeddingActivations בתרשים ה-TPU לכל היותר.
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
סטטי RecvTPUEmbeddingActivations | |
Iterator< Operand < TFloat32 >> | איטרטור () |
רשימה< פלט < TFloat32 >> | פלטים () רשימת Tensor של הפעלות הטמעה המכילה Tensor אחד לכל טבלת הטמעה במודל. |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
שיטות ציבוריות
Public static RecvTPUEmbeddingActivations create ( היקף היקף , Long numOutputs, String config)
שיטת מפעל ליצירת מחלקה העוטפת פעולת RecvTPUEmbeddingActivations חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
numoutputs | מספר טנסור הפעלת הפלט, שווה למספר טבלאות הטבעה בדגם. |
תצורה | פרוטו TPUEmbeddingConfiguration מסודר. |
החזרות
- מופע חדש של RecvTPUEmbeddingActivations