TPUReplicateMetadata

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

מטא נתונים המציינים כיצד יש לשכפל את חישוב ה-TPU.

פעולה זו מכילה את המטא-נתונים המשותפים לפעולות של תת-גרף חישוב `tpu.replicate()`.

כיתות מקוננות

מעמד TPUReplicateMetadata.Options תכונות אופציונליות עבור TPUReplicateMetadata

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

סטטי TPUReplicateMetadata.Options
allowSoftPlacement (בוליאני allowSoftPlacement)
סטטי TPUReplicateMetadata.Options
computationShape (רשימה<Long> computationShape)
סטטי TPUReplicateMetadata
צור ( היקף היקף, מספר רפליקות ארוך, אפשרויות... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת TPUReplicateMetadata חדשה.
סטטי TPUReplicateMetadata.Options
deviceAssignment (List<Long> deviceAssignment)
סטטי TPUReplicateMetadata.Options
hostComputeCore (רשימה<String> hostComputeCore)
סטטי TPUReplicateMetadata.Options
numCoresPerReplica (Long numCoresPerReplica)
סטטי TPUReplicateMetadata.Options
paddingMap (רשימה<String> paddingMap)
סטטי TPUReplicateMetadata.Options
stepMarkerLocation (מחרוזת stepMarkerLocation)
סטטי TPUReplicateMetadata.Options
טופולוגיה (טופולוגיה של מחרוזת)
סטטי TPUReplicateMetadata.Options
tpuCompileOptionsProto (מחרוזת tpuCompileOptionsProto)
סטטי TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (בוליאני useSpmdForXlaPartitioning)
סטטי TPUReplicateMetadata.Options
useTpu (useTpu בוליאני)

שיטות בירושה

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

ציבורי סטטי TPUReplicateMetadata.Options allowSoftPlacement (Boolian allowSoftPlacement)

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)

פרמטרים
הקצאת מכשיר הקצאת התקנים לחישוב.

public static TPUReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

ציבורי סטטי TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

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

public static TPUReplicateMetadata.Options paddingMap (List<String> paddingMap)

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

טופולוגיה ציבורית סטטית TPUReplicateMetadata.Options (טופולוגיית מחרוזת)

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

ציבורי סטטי TPUReplicateMetadata.Options tpuCompileOptionsProto (מחרוזת tpuCompileOptionsProto)

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

ציבורי סטטי TPUReplicateMetadata.Options useTpu (בוליאני useTpu)

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