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

NcclAllReduce

ชั้นสุดท้ายประชาชน NcclAllReduce

เอาต์พุตเทนเซอร์ที่มีการรีดิวซ์ของเทนเซอร์อินพุตทั้งหมด

ส่งออกเทนเซอร์ที่มีการลดลงในเทนเซอร์อินพุตทั้งหมดที่ส่งผ่านไปยัง ops ภายใน `shared_name เดียวกัน

กราฟควรถูกสร้างขึ้น ดังนั้นหาก op ทำงานด้วยค่า shared_name `c` ดังนั้น `num_devices` ops จะทำงานด้วยค่า shared_name `c` หากไม่ปฏิบัติตามจะทำให้การดำเนินการกราฟไม่สำเร็จ

อินพุต: อินพุตไปยังข้อมูลการลด: ค่าของการลดลงในอุปกรณ์ "num_devices" ทั้งหมด การลด: การดำเนินการลดที่จะดำเนินการ num_devices: จำนวนอุปกรณ์ที่เข้าร่วมการลดราคานี้ shared_name: ตัวระบุที่แชร์ระหว่าง ops ของการลดลงเดียวกัน

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

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

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

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

ประชาชน เอาท์พุท <T> asOutput ()

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

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

สาธารณะคง NcclAllReduce <T> สร้าง ( ขอบเขต ขอบเขต Operand <T> อินพุตลดสตริง numDevices ยาว, String sharedName)

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

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

ประชาชน เอาท์พุท <T> ข้อมูล ()