TPUOrdinalSelector

כיתת גמר ציבורית TPUOrdinalSelector

בורר ליבות TPU Op.

אופ זה מייצר קבוצה של ליבות TPU (לחימום) או ליבת TPU בודדת (להסקת הסקה רגילה) לביצוע תוכנית ה-TPU. הפלט נצרך על ידי TPUPartitionedCall.

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

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

שיטות בירושה

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

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

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

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

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

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

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

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

וקטור ליבות TPU 1 או יותר.