الطبقات
BroadcastHelper <T يمتد TType > | مشغل مساعد لإجراء عمليات البث بنمط XLA يتم بث `lhs` و`rhs` إلى نفس الترتيب، عن طريق إضافة أبعاد الحجم 1 إلى أي من `lhs` و`rhs` له رتبة أقل، باستخدام قواعد البث الخاصة بـ XLA للمشغلين الثنائيين. |
ClusterOutput <T يمتد TType > | عامل التشغيل الذي يربط مخرجات حساب XLA بعقد الرسم البياني للمستهلك الأخرى. |
التحويل <T يمتد نوع TT > | يلتف عامل التشغيل XLA ConvgeneralDilated، الموثق في https://www.tensorflow.org/performance/xla/operation_semantics#conv_convolution . |
تخلص من الكمية | يأخذ إدخال uint32 المعبأ ويفك ضغط الإدخال إلى uint8 للقيام به الإزالة على الجهاز. |
النقطة <T تمتد TType > | يلتف مشغل XLA Dotgener، الموثق في https://www.tensorflow.org/performance/xla/operation_semantics#dotgeneral . |
DynamicSlice <T يمتد TType > | يلتف مشغل XLA DynamicSlice، الموثق في https://www.tensorflow.org/performance/xla/operation_semantics#dynamicslice . |
DynamicUpdateSlice <T يمتد TType > | يلتف مشغل XLA DynamicUpdateSlice، الموثق في https://www.tensorflow.org/performance/xla/operation_semantics#dynamicupdateslice . |
اينسوم <T يمتد TType > | عملية تدعم عملية einsum الأساسية بمدخلين ومخرج واحد. |
جمع <T يمتد TType > | يلتف مشغل XLA Gather الموثق في https://www.tensorflow.org/xla/operation_semantics#gather |
KeyValueSort <T يمتد TNumber ، U يمتد TType > | يلتف عامل فرز XLA، الموثق في https://www.tensorflow.org/performance/xla/operation_semantics#sort . |
الوسادة <T تمتد TType > | يلتف مشغل XLA Pad، الموثق في https://www.tensorflow.org/performance/xla/operation_semantics#pad . |
Recv <T يمتد TType > | يتلقى الموتر المسمى من حساب XLA آخر. |
معرف النسخة المتماثلة | معرف النسخة المتماثلة. |
SelfAdjointEig <T يمتد TType > | يحسب التحلل الذاتي لمجموعة من المصفوفات المتجاورة ذاتيًا (ملاحظة: يتم دعم المدخلات الحقيقية فقط). |
يرسل | يرسل الموتر المسمى إلى حساب XLA آخر. |
المشاركة <T يمتد TType > | عملية تقوم بتجزئة الإدخال بناءً على سمة التجزئة المحددة. |
فرز <T يمتد TType > | يلتف عامل فرز XLA، الموثق في https://www.tensorflow.org/performance/xla/operation_semantics#sort . |
Svd <T يمتد TType > | يحسب التحلل الذاتي لمجموعة من المصفوفات المتجاورة ذاتيًا (ملاحظة: يتم دعم المدخلات الحقيقية فقط). |
XlaRecvFromHost <T يمتد TType > | عملية لتلقي موتر من المضيف. |
XlaSendToHost | عملية لإرسال موتر إلى المضيف. |
XlaSetBound | قم بتعيين حد لقيمة الإدخال المحددة كتلميح لمترجم Xla، ترجع نفس القيمة. |