عملیاتی که توسط پارتیشنکننده XLA SPMD برای جابجایی از پارتیشنبندی خودکار به آن استفاده میشود
پارتیشن بندی دستی ورودی را (به شکل کامل، برای پارتیشن بندی خودکار) با همان تقسیم بندی مورد استفاده در پارتیشن بندی دستی، حاشیه نویسی می کند و یک تانسور خرده شکل را خروجی می دهد تا توسط عملیات های پارتیشن بندی دستی بعدی مصرف شود. اگر شکل به طور یکنواخت قابل پارتیشن نباشد، ناحیه padding با 0 ها پوشانده می شود.
ثابت ها
رشته | OP_NAME | نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود |
روش های عمومی
خروجی <T> | asOutput () دسته نمادین تانسور را برمی گرداند. |
static <T TType > XlaSpmdFullToShardShape <T> را گسترش می دهد | |
خروجی <T> | خروجی () |
روش های ارثی
ثابت ها
رشته نهایی ثابت عمومی OP_NAME
نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود
روش های عمومی
خروجی عمومی <T> asOutput ()
دسته نمادین تانسور را برمی گرداند.
ورودی های عملیات TensorFlow خروجی های عملیات تنسورفلو دیگر هستند. این روش برای به دست آوردن یک دسته نمادین که نشان دهنده محاسبه ورودی است استفاده می شود.
ایستا عمومی XlaSpmdFullToShardShape <T> ایجاد ( دامنه دامنه ، ورودی عملوند <T>، رشته دستیSharding)
روش کارخانه برای ایجاد کلاسی که یک عملیات XlaSpmdFullToShardShape جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|
برمی گرداند
- یک نمونه جدید از XlaSpmdFullToShardShape