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