org.tensorflow.op.xla

کلاس ها

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 تنظیم کنید.

همان مقدار را برمی گرداند.