คลาสสุดท้ายสาธารณะ AllReduce
ลดเทนเซอร์หลายตัวที่มีประเภทและรูปร่างเหมือนกันร่วมกัน
คลาสที่ซ้อนกัน
| ระดับ | AllReduce.ตัวเลือก | แอ็ตทริบิวต์ทางเลือกสำหรับ AllReduce | |
ค่าคงที่
| สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
| เอาท์พุต <T> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
| AllReduce.Options แบบคงที่ | communicationHint (สตริง communicationHint) |
| คงที่ <T ขยาย TNumber > AllReduce <T> | สร้าง ( ขอบเขต ขอบเขต , ตัวดำเนินการ <T> อินพุต, ยาว groupSize, ยาว groupKey, ยาว instanceKey, สตริง mergeOp, สตริง FinalOp, รายการ <Long> subdivOffsets, ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ AllReduce ใหม่ |
| เอาท์พุต <T> | ข้อมูล () |
| AllReduce.Options แบบคงที่ | หมดเวลาวินาที (ลอยหมดเวลาวินาที) |
| AllReduce.Options แบบคงที่ | waitFor (รายการ <ยาว> รอ) |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
ค่าคงที่: "CollectiveReduce"
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สาธารณะคงที่ AllReduce <T> สร้าง (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> อินพุต, ยาว groupSize, ยาว groupKey, ยาว instanceKey, สตริง mergeOp, สตริง FinalOp, รายการ <Long> subdivOffsets, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ AllReduce ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- อินสแตนซ์ใหม่ของ AllReduce