مخرجات موتر يحتوي على التخفيض عبر جميع موترات الإدخال.
يُخرج موترًا يحتوي على التخفيض عبر جميع موترات الإدخال التي تم تمريرها إلى العمليات ضمن نفس "الاسم_المشترك".
يجب إنشاء الرسم البياني بحيث إذا تم تشغيل عملية واحدة بقيمة اسم_مشترك `c`، فسيتم تشغيل عمليات `num_devices` بقيمة اسم_مشترك `c`. سيؤدي الفشل في القيام بذلك إلى فشل إكمال تنفيذ الرسم البياني.
الإدخال: الإدخال لبيانات التخفيض: قيمة التخفيض عبر جميع أجهزة "num_devices". التخفيض: عملية التخفيض لأداء. num_devices: عدد الأجهزة المشاركة في هذا التخفيض. Shared_name: المعرف المشترك بين العمليات بنفس التخفيض.
الأساليب العامة
الإخراج <T> | كإخراج () إرجاع المقبض الرمزي للموتر. |
ثابت <T يمتد الرقم> NcclAllReduce <T> | إنشاء ( نطاق النطاق ، إدخال المعامل <T>، تقليل السلسلة، الأجهزة الطويلة، اسم السلسلة المشترك) طريقة المصنع لإنشاء فئة تغلف عملية NcclAllReduce جديدة. |
الإخراج <T> | بيانات () |
الطرق الموروثة
الأساليب العامة
الإخراج العام <T> كإخراج ()
إرجاع المقبض الرمزي للموتر.
المدخلات إلى عمليات TensorFlow هي مخرجات عملية TensorFlow أخرى. يتم استخدام هذه الطريقة للحصول على مقبض رمزي يمثل حساب الإدخال.
إنشاء NcclAllReduce <T> ثابت عام ( نطاق النطاق ، إدخال المعامل <T>، تقليل السلسلة، الأجهزة الطويلة، اسم السلسلة المشترك)
طريقة المصنع لإنشاء فئة تغلف عملية NcclAllReduce جديدة.
حدود
نِطَاق | النطاق الحالي |
---|
عائدات
- مثيل جديد من NcclAllReduce