ReplicateMetadata.Options

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

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

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

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

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

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

public ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

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

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

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

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

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

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

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

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