ReplicateMetadata

מחלקה סופית ציבורית 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 allowSoftPlacement (Boolian allowSoftPlacement)

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 hostComputeCore (רשימה<String> hostComputeCore)

Public static ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

פרמטרים
numCoresPerReplica מספר ליבות לכל העתק. משמש עבור מקביליות מודל.

public static ReplicateMetadata.Options paddingMap (רשימה<String> paddingMap)

public static ReplicateMetadata.Options stepMarkerLocation (מחרוזת stepMarkerLocation)

public static ReplicateMetadata.Options טופולוגיה (טופולוגיית מחרוזת)

פרמטרים
טופולוגיה TopologyProto המציין את הטופולוגיה של פרוסת תרמיל TPU.

public static ReplicateMetadata.Options useSpmdForXlaPartitioning (בוליאני useSpmdForXlaPartitioning)

public static ReplicateMetadata.Options useTpu (בוליאני useTpu)

פרמטרים
useTpu האם למקם את החישוב על ה-TPU.