TPUReplicateMetadata

מחלקה סופית ציבורית TPUReplicateMetadata

מטא נתונים המציינים כיצד יש לשכפל את חישוב ה-TPU.

פעולה זו מכילה את המטא-נתונים המשותפים לפעולות של תת-גרף חישוב `tpu.replicate()`.

כיתות מקוננות

מעמד

TPUReplicateMetadata.Options

תכונות אופציונליות עבור TPUReplicateMetadata

קבועים

חוּט

OP_NAME

השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

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

סטטי TPUReplicateMetadata.Options	allowSoftPlacement (בוליאני allowSoftPlacement)
סטטי TPUReplicateMetadata.Options	computationShape (רשימה<Long> computationShape)
סטטי TPUReplicateMetadata	צור ( היקף היקף, מספר רפליקות ארוך, אפשרויות... אפשרויות) שיטת מפעל ליצירת מחלקה העוטפת פעולת TPUReplicateMetadata חדשה.
סטטי 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 בוליאני)

שיטות בירושה

מכיתה org.tensorflow.op.RawOp

בוליאנית סופית	שווה (Object obj)
int סופי	hashcode ()
מבצע	אופ () החזר יחידת חישוב זו Operation אחת.
מחרוזת סופית	toString ()

מכיתה java.lang.Object

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

מתוך ממשק org.tensorflow.op.Op

אבסטרקט ExecutionEnvironment	env () החזר את סביבת הביצוע שבה נוצר האופציה הזו.
מבצע מופשט	אופ () החזר יחידת חישוב זו Operation אחת.

קבועים

מחרוזת סופית סטטית ציבורית OP_NAME

השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

ערך קבוע: "TPUReplicateMetadata"

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

ציבורי סטטי TPUReplicateMetadata.Options allowSoftPlacement (Boolian allowSoftPlacement)

public static TPUReplicateMetadata.Options computationShape (List<Long> computationShape)

פרמטרים

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

יצירת TPUReplicateMetadata סטטי ציבורי ( היקף היקף , מספר רפליקות ארוך, אפשרויות... אפשרויות)

שיטת מפעל ליצירת מחלקה העוטפת פעולת TPUReplicateMetadata חדשה.

פרמטרים

תְחוּם	ההיקף הנוכחי
numReplicas	מספר העתקים של החישוב
אפשרויות	נושא ערכי תכונות אופציונליות

החזרות

• מופע חדש של TPUReplicateMetadata

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

פרמטרים

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

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

ציבורי סטטי TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

פרמטרים

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

public static TPUReplicateMetadata.Options paddingMap (List<String> paddingMap)

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

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

פרמטרים

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

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

ציבורי סטטי TPUReplicateMetadata.Options useTpu (בוליאני useTpu)

פרמטרים

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