TPURoundRobin

שיעור הגמר הציבורי TPURoundRobin

איזון עומסים עגול על ליבות TPU.

פעולת איזון עומסים שמסתובבת בין ליבות TPU.

המבצע הזה מעביר בין המספרים השלמים ב-[0, NumTPUCoresVisiblePerHost]. זה שימושי להתממשקות עם TensorFlow אופציות שמקבלות כקלט ליבת TPU שעליה ניתן לבצע חישובים, כגון `TPUPartitionedCall`.

device_ordinal: מספר שלם ב-[0, NumTPUCoresVisiblePerHost].

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

פלט <מספר שלם>
asOutput ()
מחזירה את הידית הסמלית של טנזור.
סטטי TPURoundRobin
ליצור ( היקף היקף)
שיטת מפעל ליצירת מחלקה העוטפת פעולת TPURoundRobin חדשה.
פלט <מספר שלם>

שיטות בירושה

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

פלט ציבורי <Integer> asOutput ()

מחזירה את הידית הסמלית של טנזור.

כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.

יצירת TPURoundRobin סטטית ציבורית ( היקף היקף)

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

פרמטרים
תְחוּם ההיקף הנוכחי
החזרות
  • מופע חדש של TPURoundRobin

פלט ציבורי <Integer> deviceOrdinal ()