מחלקה סופית ציבורית TPUReplicateMetadata
מטא נתונים המציינים כיצד יש לשכפל את חישוב ה-TPU.
פעולה זו מכילה את המטא-נתונים המשותפים לפעולות של תת-גרף חישוב `tpu.replicate()`.
כיתות מקוננות
מעמד | TPUReplicateMetadata.Options | תכונות אופציונליות עבור TPUReplicateMetadata |
שיטות ציבוריות
שיטות בירושה
שיטות ציבוריות
public static TPUReplicateMetadata.Options computationShape (List<Long> computationShape)
פרמטרים
computationShape | מיושן. השתמש ב-num_cores_per_replica במקום זאת. |
---|
יצירת TPUReplicateMetadata סטטי ציבורי ( היקף היקף , מספר רפליקות ארוך, אפשרויות... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת TPUReplicateMetadata חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
numReplicas | מספר העתקים של החישוב |
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של TPUReplicateMetadata
public static TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
פרמטרים
הקצאת מכשיר | הקצאת התקנים לחישוב. |
---|
ציבורי סטטי TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
פרמטרים
numCoresPerReplica | מספר ליבות לכל העתק. משמש עבור מקביליות מודל. |
---|
טופולוגיה ציבורית סטטית TPUReplicateMetadata.Options (טופולוגיית מחרוזת)
פרמטרים
טופולוגיה | TopologyProto המציין את הטופולוגיה של פרוסת תרמיל TPU. |
---|
public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (בוליאני useSpmdForXlaPartitioning)
ציבורי סטטי TPUReplicateMetadata.Options useTpu (בוליאני useTpu)
פרמטרים
useTpu | האם למקם את החישוב על ה-TPU. |
---|