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 | 
|---|