คำเตือน: API นี้เลิกใช้งานแล้วและจะถูกลบออกใน TensorFlow เวอร์ชันในอนาคตหลังจาก การแทนที่ เสถียรแล้ว

TPURoundRobin

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

การทำโหลดบาลานซ์แบบ Round-robin บนแกน TPU

โหลดบาลานซ์ op ที่ปัดเศษระหว่างแกน TPU

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

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

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

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

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

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

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

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

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

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

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

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

เอาต์พุต สาธารณะ <จำนวนเต็ม> อุปกรณ์ลำดับ ()