מחלקה סופית ציבורית EnqueueTPUEmbeddingIntegerBatch
הפעלה שמעמידה בתור רשימה של טנזורי אצווה קלט ל-TPUEmbedding.
כיתות מקוננות
מעמד | EnqueueTPUEmbeddingIntegerBatch.Options | תכונות אופציונליות עבור EnqueueTPUEmbeddingIntegerBatch |
שיטות ציבוריות
סטטי EnqueueTPUEmbeddingIntegerBatch | צור ( היקף היקף, Iterable< Operand <Integer>> אצווה, Operand <String> modeOverride, Options... אפשרויות) שיטת מפעל ליצירת מחלקה העוטפת פעולת EnqueueTPUEmbeddingIntegerBatch חדשה. |
סטטי EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (Long deviceOrdinal) |
שיטות בירושה
שיטות ציבוריות
public static EnqueueTPUEmbeddingIntegerBatch create ( scope scope, Iterable< Operand <Integer>> batch, Operand <String> 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 במשימה שעליה ממוקם הצומת. |
---|