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