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

NcclAllReduce

המעמד הסופי הציבור NcclAllReduce

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

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

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

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

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

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

שיטות בירושה

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

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

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

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

הציבור סטטי NcclAllReduce <T> ליצור ( היקף היקף, האופרנד <T> קלט, הפחתה מחרוזת, numDevices ארוך, מחרוזת sharedName)

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

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

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