ExecuteTPUEmbeddingPartitioner

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

הפעלה שמבצעת את מחיצת ה-TPUEmbedding בתצורה המרכזית

התקן ומחשב את גודל HBM (בבתים) הנדרש לפעולת TPUEmbedding.

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

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

שיטות בירושה

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

פלט ציבורי <String> asOutput ()

מחזירה את הידית הסמלית של טנזור.

כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.

פלט ציבורי <String> commonConfig ()

פרוטו תצורה משותפת מקודדת מחרוזת המכיל מטא נתונים על פלט המחיצה TPUEmbedding וגודל ה-HBM (בבתים) הנדרש לפעולה.

public static ExecuteTPUEmbeddingPartitioner ליצור (היקף היקף , תצורת מחרוזת)

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

פרמטרים
תְחוּם ההיקף הנוכחי
תצורה פרוטו של TPUEmbeddingConfiguration מסודר למחרוזת, המתאר את תצורת ה-TPUEmbedding הרצויה.
החזרות
  • מופע חדש של ExecuteTPUEmbeddingPartitioner