TPUReplicateMetadata คลาสสุดท้ายสาธารณะ
ข้อมูลเมตาที่ระบุว่าควรจำลองการคำนวณ TPU อย่างไร
การดำเนินการนี้เก็บข้อมูลเมตาทั่วไปสำหรับการดำเนินการของกราฟย่อยการคำนวณ `tpu.replicate()`
คลาสที่ซ้อนกัน
| ระดับ | TPUReplicateMetadata ตัวเลือก | แอตทริบิวต์ที่ไม่บังคับสำหรับ TPUReplicateMetadata | |
วิธีการสาธารณะ
| TPUReplicateMetadata.Options แบบคงที่ | AllowSoftPlacement (บูลีน AllowSoftPlacement) |
| TPUReplicateMetadata.Options แบบคงที่ | computationShape (รายการ <Long> computationShape) |
| TPUReplicateMetadata แบบคงที่ | สร้าง (ขอบเขต ขอบเขต , numReplicas แบบยาว, ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUReplicateMetadata ใหม่ |
| TPUReplicateMetadata.Options แบบคงที่ | การกำหนดอุปกรณ์ (รายการ <Long> การกำหนดอุปกรณ์) |
| TPUReplicateMetadata.Options แบบคงที่ | hostComputeCore (รายการ <สตริง> โฮสต์ ComputeCore) |
| TPUReplicateMetadata.Options แบบคงที่ | numCoresPerReplica (ยาว numCoresPerReplica) |
| TPUReplicateMetadata.Options แบบคงที่ | paddingMap (รายการ <String> paddingMap) |
| TPUReplicateMetadata.Options แบบคงที่ | stepMarkerLocation (สตริง stepMarkerLocation) |
| TPUReplicateMetadata.Options แบบคงที่ | โทโพโลยี (โทโพโลยีสตริง) |
| TPUReplicateMetadata.Options แบบคงที่ | useSpmdForXlaPartitioning (การใช้บูลีน SpmdForXlaPartitioning) |
| TPUReplicateMetadata.Options แบบคงที่ | useTpu (บูลีน useTpu) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะ TPUReplicateMetadata.Options computationShape แบบคงที่ (รายการ <Long> computationShape)
พารามิเตอร์
| รูปร่างการคำนวณ | เลิกใช้แล้ว ใช้ num_cores_per_replica แทน |
|---|
การสร้าง TPUReplicateMetadata แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , numReplicas แบบยาว, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUReplicateMetadata ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| numReplicas | จำนวนการจำลองการคำนวณ |
| ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- อินสแตนซ์ใหม่ของ TPUReplicateMetadata
สาธารณะคงที่ TPUReplicateMetadata.Options อุปกรณ์การกำหนด (รายการ <ยาว> อุปกรณ์มอบหมาย)
พารามิเตอร์
| การมอบหมายอุปกรณ์ | การกำหนดอุปกรณ์สำหรับการคำนวณ |
|---|
TPUReplicateMetadata.Options สาธารณะแบบคงที่ numCoresPerReplica (Long numCoresPerReplica)
พารามิเตอร์
| numCoresPerReplica | จำนวนคอร์ต่อเรพลิกา ใช้สำหรับโมเดลความเท่าเทียม |
|---|
โทโพโลยี TPUReplicateMetadata.Options แบบคงที่สาธารณะ (โทโพโลยีสตริง)
พารามิเตอร์
| โทโพโลยี | TopologyProto ระบุโทโพโลยีของชิ้น TPU pod |
|---|
TPUReplicateMetadata.Options แบบคงที่สาธารณะ useTpu (useTpu บูลีน)
พารามิเตอร์
| ใช้Tpu | ไม่ว่าจะวางการคำนวณบน TPU |
|---|