כיתת גמר ציבורית TPUOrdinalSelector
בורר ליבות TPU Op.
אופ זה מייצר קבוצה של ליבות TPU (לחימום) או ליבת TPU בודדת (להסקת הסקה רגילה) לביצוע תוכנית ה-TPU. הפלט נצרך על ידי TPUPartitionedCall.
שיטות ציבוריות
פלט <מספר שלם> | asOutput () מחזירה את הידית הסמלית של טנזור. |
סטטי TPUOrdinalSelector | |
פלט <מספר שלם> | deviceOrdinals () וקטור ליבות TPU 1 או יותר. |
שיטות בירושה
שיטות ציבוריות
פלט ציבורי <Integer> asOutput ()
מחזירה את הידית הסמלית של טנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
יצירת TPUOrdinalSelector סטטי ציבורי ( היקף היקף)
שיטת מפעל ליצירת מחלקה העוטפת פעולת TPUOrdinalSelector חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|
החזרות
- מופע חדש של TPUOrdinalSelector