کلاس ها
BroadcastHelper <T TType را گسترش می دهد > | اپراتور کمکی برای انجام پخش به سبک XLA با استفاده از قوانین پخش XLA برای اپراتورهای باینری، «lhs» و «rhs» را با افزودن ابعاد 1 به هر یک از «lhs» و «rhs» که رتبههای پایینتری دارند، در یک رتبه پخش میکند. |
ClusterOutput <T TType > را گسترش می دهد | اپراتوری که خروجی یک محاسبات XLA را به گره های گراف مصرف کننده دیگر متصل می کند. |
تبدیل <T TType > را گسترش می دهد | اپراتور XLA ConvGeneralDilated را که در مستند شده است، می پیچد https://www.tensorflow.org/performance/xla/operation_semantics#conv_convolution. |
کوانتیزه کردن | ورودی uint32 بسته بندی شده را می گیرد و ورودی را در uint8 باز می کند تا انجام دهد کوانتیزاسیون روی دستگاه |
نقطه <T TType > را گسترش می دهد | اپراتور XLA DotGeneral را میپیچد، مستند در 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. |
Einsum <T TType > را گسترش می دهد | عملیاتی که از einsum op اولیه با 2 ورودی و 1 خروجی پشتیبانی می کند. |
جمع آوری <T extensions TType > | اپراتور XLA Gather مستند شده در را می پیچد https://www.tensorflow.org/xla/operation_semantics#gather |
KeyValueSort <T گسترش TNumber ، U گسترش TType > | عملگر XLA Sort را میپیچد، مستند در 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 دیگر دریافت می کند. |
ReplicaId | شناسه ماکت |
SelfAdjointEig <T TType را گسترش می دهد | تجزیه ویژه دسته ای از ماتریس های خود الحاقی را محاسبه می کند (توجه: فقط ورودی های واقعی پشتیبانی می شوند). |
ارسال | تانسور نامگذاری شده را به محاسبات XLA دیگری ارسال می کند. |
Sharding <T TType > را گسترش می دهد | عملیاتی که ورودی را بر اساس ویژگی اشتراک گذاری داده شده خرد می کند. |
مرتب سازی <T extensions TType > | عملگر XLA Sort را میپیچد، مستند در https://www.tensorflow.org/performance/xla/operation_semantics#sort. |
Svd <T TType را گسترش می دهد > | تجزیه ویژه دسته ای از ماتریس های خود الحاقی را محاسبه می کند (توجه: فقط ورودی های واقعی پشتیبانی می شوند). |
XlaRecvFromHost <T TType را گسترش می دهد | یک عملیات برای دریافت تانسور از میزبان. |
XlaSendToHost | گزینه ای برای ارسال یک تانسور به میزبان. |
XlaSetBound | یک کران برای مقدار ورودی داده شده به عنوان یک اشاره به کامپایلر Xla تنظیم کنید. همان مقدار را برمی گرداند. |