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