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 เพื่อทำ

การลดปริมาณบนอุปกรณ์

จุด <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 > op ที่รองรับ 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

แพด <T ขยาย TType > ล้อมตัวดำเนินการ XLA Pad ตามเอกสารที่

https://www.tensorflow.org/Performance/xla/operation_semantics#pad

รับ <T ขยาย TType > รับเทนเซอร์ที่ระบุชื่อจากการคำนวณ XLA อื่น
ReplicaId รหัสแบบจำลอง
SelfAdjointEig <T ขยาย TType > คำนวณการสลายตัวแบบไอเจนของชุดเมทริกซ์ที่อยู่ติดกันเอง

(หมายเหตุ: รองรับเฉพาะอินพุตจริงเท่านั้น)

ส่ง ส่งเทนเซอร์ที่ระบุชื่อไปยังการคำนวณ XLA อื่น
การแบ่งส่วน <T ขยาย TType > op ซึ่งแบ่งข้อมูลอินพุตตามแอตทริบิวต์การแบ่งส่วนที่กำหนด
เรียงลำดับ <T ขยาย TType > ล้อมตัวดำเนินการ XLA Sort จัดทำเอกสารไว้ที่

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

Svd <T ขยาย TType > คำนวณการสลายตัวแบบไอเจนของชุดเมทริกซ์ที่อยู่ติดกันเอง

(หมายเหตุ: รองรับเฉพาะอินพุตจริงเท่านั้น)

XlaRecvFromHost <T ขยาย TType > การดำเนินการเพื่อรับเทนเซอร์จากโฮสต์
XlaSendToHost การดำเนินการเพื่อส่งเทนเซอร์ไปยังโฮสต์
XlaSetBound ตั้งค่าขอบเขตสำหรับค่าอินพุตที่กำหนดเป็นคำใบ้ให้กับคอมไพเลอร์ Xla

ส่งกลับค่าเดียวกัน