عملية يستخدمها قسم XLA SPMD للتبديل من التقسيم التلقائي إلى
التقسيم اليدوي. فهو يوضح المدخلات (الشكل الكامل، الذي سيتم تقسيمه تلقائيًا) بنفس التجزئة المستخدمة في التقسيم اليدوي، ويخرج موترًا على شكل شظية ليتم استهلاكه من خلال عمليات مقسمة يدويًا لاحقًا. إذا لم يكن الشكل قابلاً للتقسيم بشكل متساوٍ، فسيتم إخفاء منطقة الحشو بالصفر.
الثوابت
خيط | OP_NAME | اسم هذه العملية كما هو معروف بواسطة محرك TensorFlow الأساسي |
الأساليب العامة
الإخراج <T> | كإخراج () إرجاع المقبض الرمزي للموتر. |
ثابت <T يمتد TType > XlaSpmdFullToShardShape <T> | |
الإخراج <T> | انتاج () |
الطرق الموروثة
الثوابت
السلسلة النهائية الثابتة العامة OP_NAME
اسم هذه العملية كما هو معروف بواسطة محرك TensorFlow الأساسي
الأساليب العامة
الإخراج العام <T> كإخراج ()
إرجاع المقبض الرمزي للموتر.
المدخلات إلى عمليات TensorFlow هي مخرجات عملية TensorFlow أخرى. يتم استخدام هذه الطريقة للحصول على مقبض رمزي يمثل حساب الإدخال.
إنشاء XlaSpmdFullToShardShape الثابت العام <T> (نطاق النطاق ، وإدخال المعامل <T>، وسلسلة ManualSharding)
طريقة المصنع لإنشاء فئة تلتف حول عملية XlaSpmdFullToShardShape جديدة.
حدود
نِطَاق | النطاق الحالي |
---|
عائدات
- مثيل جديد لـ XlaSpmdFullToShardShape