מחלקה סטטית ציבורית ReplicateMetadata.Options
תכונות אופציונליות עבור ReplicateMetadata
שיטות ציבוריות
ReplicateMetadata.Options | allowSoftPlacement (בוליאני allowSoftPlacement) |
ReplicateMetadata.Options | computationShape (רשימה<Long> computationShape) |
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 בוליאני) |
שיטות בירושה
שיטות ציבוריות
public ReplicateMetadata.Options computationShape (רשימה<Long> computationShape)
פרמטרים
computationShape | מיושן. השתמש ב-num_cores_per_replica במקום זאת. |
---|
public ReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
פרמטרים
הקצאת מכשיר | הקצאת התקנים לחישוב. |
---|
public ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
פרמטרים
numCoresPerReplica | מספר ליבות לכל העתק. משמש עבור מקביליות מודל. |
---|
public ReplicateMetadata.Options טופולוגיה (טופולוגיית מחרוזת)
פרמטרים
טופולוגיה | TopologyProto המציין את הטופולוגיה של פרוסת תרמיל TPU. |
---|
public ReplicateMetadata.Options useTpu (בוליאני useTpu)
פרמטרים
useTpu | האם למקם את החישוב על ה-TPU. |
---|