איזון עומסים עגול על ליבות TPU.
פעולת איזון עומסים שמסתובבת בין ליבות TPU.
המבצע הזה מעביר בין המספרים השלמים ב-[0, NumTPUCoresVisiblePerHost]. זה שימושי להתממשקות עם TensorFlow אופציות שמקבלות כקלט ליבת TPU שעליה ניתן לבצע חישובים, כגון `TPUPartitionedCall`.
device_ordinal: מספר שלם ב-[0, NumTPUCoresVisiblePerHost].
שיטות ציבוריות
פלט <מספר שלם> | asOutput () מחזירה את הידית הסמלית של טנזור. |
סטטי TPURoundRobin | |
פלט <מספר שלם> |
שיטות בירושה
שיטות ציבוריות
פלט ציבורי <Integer> asOutput ()
מחזירה את הידית הסמלית של טנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
יצירת TPURoundRobin סטטית ציבורית ( היקף היקף)
שיטת מפעל ליצירת מחלקה העוטפת פעולת TPURoundRobin חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|
החזרות
- מופע חדש של TPURoundRobin