عملیاتی که فعالسازیهای تعبیهشده را روی TPU دریافت میکند.
سیستم TPU جستجوهای جاسازی و تجمیع را انجام می دهد. نتایج این تجمیعها برای نمودار Tensorflow بهعنوان خروجیهای یک XlaRecvTPUEmbeddingActivations عملیات قابل مشاهده است. این عملیات فهرستی حاوی یک تانسور فعال سازی در هر جدول مشخص شده در مدل را برمی گرداند.
روش های عمومی
استاتیک XlaRecvTPUEmbeddingActivations | |
Iterator< عملوند <Float>> | اشاره گر () |
فهرست< خروجی <شناور>> | خروجی ها () TensorList از فعالسازیهای تعبیهشده حاوی یک تانسور در هر جدول جاسازی در مدل. |
روش های ارثی
روش های عمومی
استاتیک عمومی XlaRecvTPUEmbeddingActivations ایجاد ( حوزه دامنه ، عملوند <?> deduplicationData، NumTables طولانی، پیکربندی رشته)
روش کارخانه برای ایجاد کلاسی که یک عملیات XlaRecvTPUEmbeddingActivations جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
DeduplicationData | یک تانسور با نوع=DT_VARIANT که حاوی دادههای تکراری است. تانسور یک تاپل تو در تو XLA حاوی N عنصر است (که در آن N نسبت تعداد هسته های جاسازی شده به هسته های تانسور در هر تراشه TPU است). هر عنصر از تاپل تودرتو یک تاپل از تانسورهای رتبه 1 است. هر تانسور یا شامل شاخصهایی (DT_UINT32) برای جاسازی جستجو در TensorCore یا وزنهایی (DT_FLOAT) برای اعمال در خروجی عملیات جستجوی جاسازی است. |
numTables | تعداد تانسورهای فعال سازی خروجی. اگر توصیفگر ویژگی در پیکربندی جاسازی tpu وجود داشته باشد، برابر با تعداد ویژگیها در غیر این صورت برابر با تعداد جداول جاسازی در مدل است. |
پیکربندی | پروتوی TPUEmbeddingConfiguration سریال. |
برمی گرداند
- یک نمونه جدید از XlaRecvTPUEmbeddingActivations
لیست عمومی< خروجی <شناور>> خروجی ()
TensorList از فعالسازیهای تعبیهشده حاوی یک تانسور در هر جدول جاسازی در مدل.