NcclAllReduce

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

מוציא טנזור המכיל את ההפחתה על פני כל טנסור הקלט.

מוציא טנסור המכיל את ההפחתה על פני כל טנסור הקלט שהועברו ל-ops בתוך אותו `shared_name.

הגרף צריך להיות בנוי כך שאם הפעלה אחת פועלת עם ערך shared_name `c`, אז פעולות של `num_devices` יפעלו עם ערך shared_name `c`. אי ביצוע פעולה זו יגרום לכשל בביצוע הגרף.

קלט: הקלט לנתוני ההפחתה: ערך ההפחתה בכל מכשירי `num_devices`. צמצום: פעולת ההפחתה שיש לבצע. num_devices: מספר המכשירים המשתתפים בהפחתה זו. shared_name: מזהה ששיתף בין פעולות של אותה הפחתה.

קבועים

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

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

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

שיטות בירושה

קבועים

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

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

ערך קבוע: "NcclAllReduce"

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

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

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

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

public static NcclAllReduce <T> create ( היקף היקף , קלט Operand <T>, הפחתת מחרוזת, מספר מכשירים ארוכים, מחרוזת משותף שם)

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

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

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