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