org.tensorflow.op.xla

שיעורים

BroadcastHelper <T מרחיב את TType > מפעיל עוזר לביצוע שידורים בסגנון XLA

משדר `lhs` ו-`rhs` לאותה דרגה, על-ידי הוספת ממדים בגודל 1 למי מבין `lhs` ו-`rhs` בעל הדרגה הנמוכה יותר, באמצעות כללי השידור של XLA עבור אופרטורים בינאריים.

ClusterOutput <T מרחיב את TType > מפעיל המחבר את הפלט של חישוב XLA לצמתי גרף צרכנים אחרים.
Conv <T מרחיב את TType > עוטף את האופרטור XLA ConvGeneralDilated, מתועד ב

https://www.tensorflow.org/performance/xla/operation_semantics#conv_convolution .

דהקוונטיזציה לוקח את קלט uint32 ארוז ומפרק את הקלט ל-uint8 לעשות

דקוונטיזציה במכשיר.

Dot <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 מרחיב את 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 .

Pad <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 מרחיב את TType > עוטף את מפעיל XLA Sort, מתועד ב

https://www.tensorflow.org/performance/xla/operation_semantics#sort .

Svd <T מרחיב את TType > מחשב את הפירוק העצמי של אצווה של מטריצות צמודות עצמיות

(הערה: רק קלט אמיתי נתמך).

XlaRecvFromHost <T מרחיב את TType > אופ לקבלת טנזור מהמארח.
XlaSendToHost אופציה לשלוח טנזור למארח.
XlaSetBound הגדר גבול לערך הקלט הנתון כרמז למהדר Xla,

מחזיר את אותו הערך.