ประชาชน InfeedEnqueueTuple ชั้นสุดท้าย
ป้อนค่าเทนเซอร์หลายค่าในการคำนวณเป็นทูเพิล XLA
คลาสที่ซ้อนกัน
ระดับ | InfeedEnqueueTuple.Options | คุณลักษณะที่จำเป็นสำหรับการ InfeedEnqueueTuple |
วิธีการสาธารณะ
คง InfeedEnqueueTuple | สร้าง ( ขอบเขต ขอบเขต Iterable < Operand <? >> ปัจจัยการผลิตรายการ < รูปร่าง > รูปร่าง ตัวเลือก ... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ InfeedEnqueueTuple ใหม่ |
คง InfeedEnqueueTuple.Options | deviceOrdinal (Long deviceOrdinal) |
คง InfeedEnqueueTuple.Options | รูปแบบ (รายชื่อ <ยาว> รูปแบบ) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง InfeedEnqueueTuple สร้าง ( ขอบเขต ขอบเขต Iterable < Operand <? >> ปัจจัยการผลิตรายการ < รูปร่าง > รูปร่าง ตัวเลือก ... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ InfeedEnqueueTuple ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
อินพุต | รายการเทนเซอร์ที่จะจัดเตรียมโดยใช้กลไกการป้อนเข้า |
รูปร่าง | รูปร่างของแต่ละเทนเซอร์ใน "อินพุต" |
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ InfeedEnqueueTuple
สาธารณะคง InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)
พารามิเตอร์
อุปกรณ์ลำดับ | อุปกรณ์ TPU ที่จะใช้ ซึ่งควรเป็น -1 เมื่อ Op ทำงานบนอุปกรณ์ TPU และ >= 0 เมื่อ Op ทำงานบนอุปกรณ์ CPU |
---|
สาธารณะคง InfeedEnqueueTuple.Options รูปแบบ (รายชื่อ <ยาว> รูปแบบ)
พารามิเตอร์
เลย์เอาต์ | เวกเตอร์ถือเค้าโครงที่ร้องขอในลำดับรองถึงหลักสำหรับรูปร่างทูเปิลทั้งหมด ตามลำดับ รูปร่างปรากฏในอินพุต "รูปร่าง" องค์ประกอบโครงร่างสำหรับรูปร่างย่อยสามารถตั้งค่าเป็น -1 ซึ่งในกรณีนี้ โครงร่างที่เกี่ยวข้องจะถูกคำนวณโดยการดำเนินการป้อนเข้า |
---|