TPUReplicateMetadata คลาสสุดท้ายสาธารณะ
ข้อมูลเมตาที่ระบุว่าควรจำลองการคำนวณ TPU อย่างไร
การดำเนินการนี้เก็บข้อมูลเมตาทั่วไปสำหรับการดำเนินการของกราฟย่อยการคำนวณ `tpu.replicate()`
คลาสที่ซ้อนกัน
ระดับ | TPUReplicateMetadata ตัวเลือก | แอตทริบิวต์ที่ไม่บังคับสำหรับ TPUReplicateMetadata |
ค่าคงที่
สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
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) |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
ค่าคงที่: "TPUReplicateMetadata"
วิธีการสาธารณะ
สาธารณะ 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 |
---|