XlaSpmdShardToFullShape

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

אופציה המשמשת את מחיצת XLA SPMD למעבר מחלוקה ידנית ל

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

קבועים

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

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

פלט <T>
asOutput ()
מחזירה את הידית הסמלית של הטנזור.
סטטי <T מרחיב את TType > XlaSpmdShardToFullShape <T>
צור ( היקף היקף , קלט Operand <T>, String manualSharding, Shape fullShape)
שיטת מפעל ליצירת מחלקה העוטפת פעולת XlaSpmdShardToFullShape חדשה.
פלט <T>
פלט ()

שיטות בירושה

קבועים

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

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

ערך קבוע: "XlaSpmdShardToFullShape"

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

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

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

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

XlaSpmdShardToFullShape <T> יצירת סטטי ציבורי ( scope scope, Operand <T> קלט, String manualSharding, Shape fullShape)

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

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

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