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