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