NcclAllReduce

کلاس نهایی عمومی NcclAllReduce

خروجی یک تانسور حاوی کاهش در تمام تانسورهای ورودی.

یک تانسور حاوی کاهش در تمام تانسورهای ورودی ارسال شده به عملیات در همان «نام_اشتراک‌شده» را خروجی می‌دهد.

نمودار باید طوری ساخته شود که اگر یکی از عملیات ها با مقدار shared_name «c» اجرا شود، عملیات «num_devices» با مقدار shared_name «c» اجرا شود. عدم انجام این کار باعث می شود که اجرای گراف کامل نشود.

ورودی: ورودی داده‌های کاهش: مقدار کاهش در همه دستگاه‌های «تعداد_دستگاه‌ها». کاهش: عملیات کاهش برای انجام. num_devices: تعداد دستگاه های شرکت کننده در این کاهش. shared_name: شناسه ای که بین عملیات های مشابه کاهش به اشتراک گذاشته شده است.

ثابت ها

رشته OP_NAME نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود

روش های عمومی

خروجی <T>
asOutput ()
دسته نمادین تانسور را برمی گرداند.
استاتیک <T TNumber > NcclAllReduce <T> را گسترش می دهد
ایجاد (حوزه دامنه ، ورودی عملوند <T>، کاهش رشته، تعداد زیاد دستگاه‌ها، رشته sharedName)
روش کارخانه برای ایجاد کلاسی که یک عملیات NcclAllReduce جدید را بسته بندی می کند.
خروجی <T>

روش های ارثی

ثابت ها

رشته نهایی ثابت عمومی OP_NAME

نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود

مقدار ثابت: "NcclAllReduce"

روش های عمومی

خروجی عمومی <T> asOutput ()

دسته نمادین تانسور را برمی گرداند.

ورودی های عملیات TensorFlow خروجی های عملیات تنسورفلو دیگر هستند. این روش برای به دست آوردن یک دسته نمادین که نشان دهنده محاسبه ورودی است استفاده می شود.

عمومی ثابت NcclAllReduce <T> ایجاد ( دامنه دامنه ، ورودی عملوند <T>، کاهش رشته، تعداد زیاد دستگاه‌ها، رشته sharedName)

روش کارخانه برای ایجاد کلاسی که یک عملیات NcclAllReduce جدید را بسته بندی می کند.

مولفه های
محدوده محدوده فعلی
برمی گرداند
  • یک نمونه جدید از NcclAllReduce

اطلاعات خروجی عمومی <T> ()