ExecuteTPUEmbeddingPartitioner

الفئة النهائية العامة ExecuteTPUEmbeddingPartitioner

عملية تنفذ مقسم TPUEmbedding على التكوين المركزي

الجهاز ويحسب حجم HBM (بالبايت) المطلوب لعملية TPUEmbedding.

الأساليب العامة

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

الطرق الموروثة

الأساليب العامة

الإخراج العام <سلسلة> كإخراج ()

إرجاع المقبض الرمزي للموتر.

المدخلات إلى عمليات TensorFlow هي مخرجات عملية TensorFlow أخرى. يتم استخدام هذه الطريقة للحصول على مقبض رمزي يمثل حساب الإدخال.

الإخراج العام <سلسلة> CommonConfig ()

نموذج تكوين مشترك مشفر بسلسلة يحتوي على بيانات تعريف حول مخرجات مقسم TPUEmbedding وحجم HBM (بالبايت) المطلوب للتشغيل.

إنشاء ExecuteTPUEmbeddingPartitioner الثابت العام (نطاق النطاق ، تكوين السلسلة)

طريقة المصنع لإنشاء فئة تغلف عملية ExecuteTPUEmbeddingPartitioner جديدة.

حدود
نِطَاق النطاق الحالي
التكوين نموذج أولي لـ TPUEmbeddingConfiguration متسلسل إلى سلسلة، يصف تكوين TPUEmbedding المطلوب.
عائدات
  • مثيل جديد لـ ExecuteTPUEmbeddingPartitioner