TPURoundRobin

TPURoundRobin คลาสสุดท้ายสาธารณะ

การปรับสมดุลโหลดแบบ Round-robin บนแกน TPU

การทำโหลดบาลานซ์แบบวนรอบระหว่างแกน TPU

การปัดเศษ op นี้ระหว่างจำนวนเต็มใน [0, NumTPUCoresVisiblePerHost] มีประโยชน์สำหรับการเชื่อมต่อกับการดำเนินการของ TensorFlow ที่ใช้เป็นอินพุตแกน TPU เพื่อใช้ประมวลผลการคำนวณ เช่น `TPUPartitionedCall`

device_ordinal: จำนวนเต็มใน [0, NumTPUCoresVisiblePerHost]

วิธีการสาธารณะ

เอาต์พุต <จำนวนเต็ม>
เป็นเอาท์พุต ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
TPURoundRobin แบบคงที่
สร้าง ( ขอบเขต ขอบเขต)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ TPURoundRobin ใหม่
เอาต์พุต <จำนวนเต็ม>
อุปกรณ์ลำดับ ()

วิธีการสืบทอด

วิธีการสาธารณะ

เอาท์พุท สาธารณะ <จำนวนเต็ม> asOutput ()

ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์

อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต

การสร้าง TPURoundRobin แบบคงที่สาธารณะ (ขอบเขต ขอบเขต )

วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ TPURoundRobin ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
การส่งคืน
  • อินสแตนซ์ใหม่ของ TPURoundRobin

สาธารณะ เอาท์พุท <จำนวนเต็ม> deviceOrdinal ()