خروجی یک تانسور حاوی کاهش در تمام تانسورهای ورودی.
یک تانسور حاوی کاهش در تمام تانسورهای ورودی ارسال شده به عملیات در همان «نام_اشتراک‌شده» را خروجی می‌دهد.
نمودار باید طوری ساخته شود که اگر یکی از عملیات ها با مقدار shared_name «c» اجرا شود، عملیات «num_devices» با مقدار shared_name «c» اجرا شود. عدم انجام این کار باعث می شود که اجرای گراف کامل نشود.
ورودی: ورودی داده‌های کاهش: مقدار کاهش در همه دستگاه‌های «تعداد_دستگاه‌ها». کاهش: عملیات کاهش برای انجام. num_devices: تعداد دستگاه های شرکت کننده در این کاهش. shared_name: شناسه ای که بین عملیات های مشابه کاهش به اشتراک گذاشته شده است.
ثابت ها
| رشته | OP_NAME | نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود |
روش های عمومی
| خروجی <T> | asOutput () دسته نمادین تانسور را برمی گرداند. |
| استاتیک <T TNumber > NcclAllReduce <T> را گسترش می دهد | |
| خروجی <T> | داده ها () |
روش های ارثی
ثابت ها
رشته نهایی ثابت عمومی OP_NAME
نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود
روش های عمومی
خروجی عمومی <T> asOutput ()
دسته نمادین تانسور را برمی گرداند.
ورودی های عملیات TensorFlow خروجی های عملیات تنسورفلو دیگر هستند. این روش برای به دست آوردن یک دسته نمادین که نشان دهنده محاسبه ورودی است استفاده می شود.
عمومی ثابت NcclAllReduce <T> ایجاد ( دامنه دامنه ، ورودی عملوند <T>، کاهش رشته، تعداد زیاد دستگاه‌ها، رشته sharedName)
روش کارخانه برای ایجاد کلاسی که یک عملیات NcclAllReduce جدید را بسته بندی می کند.
مولفه های
| محدوده | محدوده فعلی |
|---|
برمی گرداند
- یک نمونه جدید از NcclAllReduce