ลด "อินพุต" จาก "num_devices" โดยใช้ "การลด" ลงในอุปกรณ์เครื่องเดียว
ลด "อินพุต" จาก "num_devices" โดยใช้ "การลด" ลงในอุปกรณ์เครื่องเดียว
ควรสร้างกราฟเพื่อให้อินพุตทั้งหมดมีการกำหนดอุปกรณ์ที่ถูกต้อง และ op เองก็ถูกกำหนดหนึ่งในอุปกรณ์เหล่านี้
อินพุต: อินพุตเพื่อลด ข้อมูล: ค่าที่ลดลงในอุปกรณ์ `num_devices` ทั้งหมด การลดลง: การดำเนินการลดที่จะดำเนินการ
ค่าคงที่
สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
เอาท์พุต <T> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
คง <T ขยาย TNumber > NcclReduce <T> | |
เอาท์พุต <T> | ข้อมูล () |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สร้าง NcclReduce <T> แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , อินพุต < ตัวดำเนินการ <T>>, การลดสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ NcclReduce ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|
การส่งคืน
- อินสแตนซ์ใหม่ของ NcclReduce