ประชาชน EnqueueTPUEmbeddingIntegerBatch ชั้นสุดท้าย
op ที่จัดคิวรายการของแบทช์เทนเซอร์อินพุตไปยัง TPUEmbedding
คลาสที่ซ้อนกัน
ระดับ | EnqueueTPUEmbeddingIntegerBatch.Options | คุณลักษณะที่จำเป็นสำหรับการ EnqueueTPUEmbeddingIntegerBatch |
วิธีการสาธารณะ
คง EnqueueTPUEmbeddingIntegerBatch | สร้าง ( ขอบเขต ขอบเขต Iterable < Operand <จำนวนเต็ม >> ชุด Operand <String> modeOverride, ตัวเลือก ... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ EnqueueTPUEmbeddingIntegerBatch ใหม่ |
คง EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (Long deviceOrdinal) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง EnqueueTPUEmbeddingIntegerBatch สร้าง ( ขอบเขต ขอบเขต Iterable < Operand <จำนวนเต็ม >> ชุด Operand <String> modeOverride, ตัวเลือก ... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ EnqueueTPUEmbeddingIntegerBatch ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ชุด | รายการเทนเซอร์ 1 มิติ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งประกอบด้วยดัชนีลงในตาราง |
modeOverride | อินพุตสตริงที่แทนที่โหมดที่ระบุใน TPUEmbeddingConfiguration ค่าที่รองรับคือ {'unspecified', 'inference', 'training', 'backward_pass_only'} เมื่อตั้งค่าเป็น 'ไม่ระบุ' โหมดที่ตั้งค่าใน TPUEmbeddingConfiguration จะถูกใช้ มิฉะนั้น mode_override จะถูกใช้ |
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ EnqueueTPUEmbeddingIntegerBatch
สาธารณะคง EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
พารามิเตอร์
อุปกรณ์ลำดับ | อุปกรณ์ TPU ที่จะใช้ ควรเป็น >= 0 และน้อยกว่าจำนวนคอร์ TPU ในงานที่วางโหนด |
---|