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