عملية تنفذ مقسم TPUEmbedding على التكوين المركزي
الجهاز ويحسب حجم HBM (بالبايت) المطلوب لعملية TPUEmbedding.
الأساليب العامة
الإخراج <سلسلة> | كإخراج () إرجاع المقبض الرمزي للموتر. |
الإخراج <سلسلة> | التكوين المشترك () نموذج تكوين مشترك مشفر بسلسلة يحتوي على بيانات تعريف حول مخرجات مقسم TPUEmbedding وحجم HBM (بالبايت) المطلوب للتشغيل. |
تنفيذ ثابتTPUEmbeddingPartitioner |
الطرق الموروثة
الأساليب العامة
الإخراج العام <سلسلة> كإخراج ()
إرجاع المقبض الرمزي للموتر.
المدخلات إلى عمليات TensorFlow هي مخرجات عملية TensorFlow أخرى. يتم استخدام هذه الطريقة للحصول على مقبض رمزي يمثل حساب الإدخال.
الإخراج العام <سلسلة> CommonConfig ()
نموذج تكوين مشترك مشفر بسلسلة يحتوي على بيانات تعريف حول مخرجات مقسم TPUEmbedding وحجم HBM (بالبايت) المطلوب للتشغيل.
إنشاء ExecuteTPUEmbeddingPartitioner الثابت العام (نطاق النطاق ، تكوين السلسلة)
طريقة المصنع لإنشاء فئة تغلف عملية ExecuteTPUEmbeddingPartitioner جديدة.
حدود
نِطَاق | النطاق الحالي |
---|---|
التكوين | نموذج أولي لـ TPUEmbeddingConfiguration متسلسل إلى سلسلة، يصف تكوين TPUEmbedding المطلوب. |
عائدات
- مثيل جديد لـ ExecuteTPUEmbeddingPartitioner