מחלקה סופית ציבורית EnqueueTPUEmbeddingIntegerBatch
הפעלה שמעמידה בתור רשימה של טנזורי אצווה קלט ל-TPUEmbedding.
כיתות מקוננות
מעמד | EnqueueTPUEmbeddingIntegerBatch.Options | תכונות אופציונליות עבור EnqueueTPUEmbeddingIntegerBatch |
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
סטטי EnqueueTPUEmbeddingIntegerBatch | |
סטטי EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (Long deviceOrdinal) |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
ערך קבוע: "EnqueueTPUEmbeddingIntegerBatch"
שיטות ציבוריות
public static EnqueueTPUEmbeddingIntegerBatch create ( scope scope, Iterable< Operand < TInt32 >> batch, Operand < TString > modeOverride, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת EnqueueTPUEmbeddingIntegerBatch חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
קבוצה | רשימה של טנזורים 1D, אחד לכל טבלת הטבעה, המכילה את המדדים לטבלאות. |
מצב ביטול | קלט מחרוזת שעוקף את המצב שצוין ב-TPUEmbeddingConfiguration. הערכים הנתמכים הם {'unspecificed', 'inference', 'training', 'backward_pass_only'}. כאשר מוגדר ל'unspecificed', נעשה שימוש במצב שהוגדר ב-TPUEmbeddingConfiguration, אחרת נעשה שימוש ב-mode_override. |
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של EnqueueTPUEmbeddingIntegerBatch
public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
פרמטרים
deviceOrdinal | מכשיר ה-TPU לשימוש. צריך להיות >= 0 וקטן ממספר ליבות ה-TPU במשימה שעליה ממוקם הצומת. |
---|