ชั้นเรียน
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 ส่งกลับค่าเดียวกัน |