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 בוליאני)

שיטות בירושה

מכיתה java.lang.Object

בוליאני	שווה (Object arg0)
שיעור אחרון<?>	getClass ()
int	hashcode ()
ריק סופי	להודיע ​​()
ריק סופי	הודע הכל ()
חוּט	toString ()
ריק סופי	המתן (ארג0 ארוך, int arg1)
ריק סופי	המתן (ארג0 ארוך)
ריק סופי	חכה ()

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

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.