XlaSpmdFullToShardShape

כיתה גמר ציבורית XlaSpmdFullToShardShape

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

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

קבועים

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

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

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

שיטות בירושה

קבועים

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

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

ערך קבוע: "XlaSpmdFullToShardShape"

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

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

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

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

סטטי ציבורי XlaSpmdFullToShardShape <T> create ( היקף היקף , קלט Operand <T>, String manualSharding)

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

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

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