การปรับสมดุลโหลดแบบ Round-robin บนแกน TPU
การทำโหลดบาลานซ์แบบวนรอบระหว่างแกน TPU
การปัดเศษ op นี้ระหว่างจำนวนเต็มใน [0, NumTPUCoresVisiblePerHost] มีประโยชน์สำหรับการเชื่อมต่อกับการดำเนินการของ TensorFlow ที่ใช้เป็นอินพุตแกน TPU เพื่อใช้ประมวลผลการคำนวณ เช่น `TPUPartitionedCall`
device_ordinal: จำนวนเต็มใน [0, NumTPUCoresVisiblePerHost]
วิธีการสาธารณะ
เอาต์พุต <จำนวนเต็ม> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
TPURoundRobin แบบคงที่ | |
เอาต์พุต <จำนวนเต็ม> | อุปกรณ์ลำดับ () |
วิธีการสืบทอด
วิธีการสาธารณะ
เอาท์พุท สาธารณะ <จำนวนเต็ม> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
การสร้าง TPURoundRobin แบบคงที่สาธารณะ (ขอบเขต ขอบเขต )
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ TPURoundRobin ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|
การส่งคืน
- อินสแตนซ์ใหม่ของ TPURoundRobin