מפחית 'קלט' מ-'num_devices' באמצעות 'reduction' למכשיר בודד.
מפחית 'קלט' מ-'num_devices' באמצעות 'reduction' למכשיר בודד.
הגרף צריך להיות בנוי כך שלכל התשומות תהיה הקצאת התקן חוקית, ולאופ עצמו מוקצה אחד מההתקנים הללו.
קלט: הקלט להפחתה. נתונים: ערך ההפחתה בכל מכשירי `num_devices`. צמצום: פעולת ההפחתה שיש לבצע.
קבועים
חוּט | OP_NAME | השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow |
שיטות ציבוריות
פלט <T> | asOutput () מחזירה את הידית הסמלית של הטנזור. |
סטטי <T מרחיב TNumber > NcclReduce <T> | |
פלט <T> | נתונים () |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית OP_NAME
השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow
שיטות ציבוריות
פלט ציבורי <T> asOutput ()
מחזירה את הידית הסמלית של הטנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
public static NcclReduce <T> create ( היקף היקף , קלט איטרני< Oprand <T>>, הפחתת מחרוזת)
שיטת מפעל ליצירת מחלקה העוטפת פעולת NcclReduce חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|
החזרות
- מופע חדש של NcclReduce