TPUReplicateMetadata.Options

מחלקה סטטית ציבורית 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
useSpmdForXlaPartitioning (בוליאני useSpmdForXlaPartitioning)
TPUReplicateMetadata.Options
useTpu (useTpu בוליאני)

שיטות בירושה

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

public TPUReplicateMetadata.Options allowSoftPlacement (Boolian allowSoftPlacement)

public TPUReplicateMetadata.Options computationShape (רשימה<Long> computationShape)

פרמטרים
computationShape מיושן. השתמש ב-num_cores_per_replica במקום זאת.

public TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)

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

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

public TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

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

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

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

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

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

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

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

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