کلاس نهایی عمومی EnqueueTPUEmbeddingIntegerBatch
عملیاتی که لیستی از تانسورهای دسته ای ورودی را در TPUEmbedding قرار می دهد.
کلاس های تو در تو
کلاس | EnqueueTPUEmbeddingIntegerBatch.Options | ویژگی های اختیاری برای EnqueueTPUEmbeddingIntegerBatch |
ثابت ها
رشته | OP_NAME | نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود |
روش های عمومی
استاتیک EnqueueTPUEmbeddingIntegerBatch | |
استاتیک EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (Long deviceOrdinal) |
روش های ارثی
ثابت ها
رشته نهایی ثابت عمومی OP_NAME
نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود
مقدار ثابت: "EnqueueTPUEmbeddingIntegerBatch"
روش های عمومی
عمومی استاتیک EnqueueTPUEmbeddingIntegerBatch ایجاد ( دامنه دامنه ، تکرارپذیر< Operand < TINT32 >> دسته، عملوند < TString > modeOverride، گزینهها... گزینهها)
روش کارخانه برای ایجاد کلاسی که یک عملیات EnqueueTPUEmbeddingIntegerBatch جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
دسته ای | فهرستی از تانسورهای یک بعدی، یکی برای هر جدول تعبیه شده، که شامل شاخص های جداول است. |
modeOverride | ورودی رشته ای که حالت مشخص شده در TPUEmbeddingConfiguration را لغو می کند. مقادیر پشتیبانی شده عبارتند از: {'نامشخص'، 'استنتاج'، 'training'، 'backward_pass_only'}. وقتی روی "نا مشخص" تنظیم شود، حالت تنظیم شده در TPUEmbeddingConfiguration استفاده می شود، در غیر این صورت mode_override استفاده می شود. |
گزینه ها | مقادیر ویژگی های اختیاری را حمل می کند |
برمی گرداند
- یک نمونه جدید از EnqueueTPUEmbeddingIntegerBatch
عمومی استاتیک EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
مولفه های
دستگاه عادی | دستگاه TPU برای استفاده. باید >= 0 و کمتر از تعداد هسته های TPU در کاری که گره روی آن قرار می گیرد باشد. |
---|