הפעלה שמבצעת את מחיצת ה-TPUEmbedding בתצורה המרכזית
התקן ומחשב את גודל HBM (בבתים) הנדרש לפעולת TPUEmbedding.
שיטות ציבוריות
פלט <מחרוזת> | asOutput () מחזירה את הידית הסמלית של טנזור. |
פלט <מחרוזת> | commonConfig () פרוטו תצורה משותפת מקודדת מחרוזת המכיל מטא נתונים על פלט המחיצה TPUEmbedding וגודל ה-HBM (בבתים) הנדרש לפעולה. |
סטטי ExecuteTPUEmbeddingPartitioner |
שיטות בירושה
שיטות ציבוריות
פלט ציבורי <String> asOutput ()
מחזירה את הידית הסמלית של טנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
פלט ציבורי <String> commonConfig ()
פרוטו תצורה משותפת מקודדת מחרוזת המכיל מטא נתונים על פלט המחיצה TPUEmbedding וגודל ה-HBM (בבתים) הנדרש לפעולה.
public static ExecuteTPUEmbeddingPartitioner ליצור (היקף היקף , תצורת מחרוזת)
שיטת מפעל ליצירת מחלקה העוטפת פעולת ExecuteTPUEmbeddingPartitioner חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
תצורה | פרוטו של TPUEmbeddingConfiguration מסודר למחרוזת, המתאר את תצורת ה-TPUEmbedding הרצויה. |
החזרות
- מופע חדש של ExecuteTPUEmbeddingPartitioner