כיתת גמר ציבורית XlaSpmdShardToFullShape
אופציה המשמשת את מחיצת XLA SPMD למעבר מחלוקה ידנית ל
חלוקה אוטומטית. הוא ממיר את הקלט בצורת רסיס, המחולק באופן ידני, לטנזור בצורת מלא כדי להיות מחולק אוטומטית עם אותו ריסוק המשמש לחלוקה ידנית.
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
פלט <T> | asOutput () מחזירה את הידית הסמלית של הטנזור. |
סטטי <T מרחיב את TType > XlaSpmdShardToFullShape <T> | |
פלט <T> | פלט () |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
ערך קבוע: "XlaSpmdShardToFullShape"
שיטות ציבוריות
פלט ציבורי <T> asOutput ()
מחזירה את הידית הסמלית של הטנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
XlaSpmdShardToFullShape <T> יצירת סטטי ציבורי ( scope scope, Operand <T> קלט, String manualSharding, Shape fullShape)
שיטת מפעל ליצירת מחלקה העוטפת פעולת XlaSpmdShardToFullShape חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|
החזרות
- מופע חדש של XlaSpmdShardToFullShape