ReplicateMetadata

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 คง ที่อนุญาตให้ SoftPlacement (บูลีน AllowSoftPlacement)

สาธารณะ ReplicateMetadata.Options computationShape แบบคงที่ (รายการ <Long> computationShape)

พารามิเตอร์
รูปร่างการคำนวณ เลิกใช้แล้ว ใช้ num_cores_per_replica แทน

การสร้าง ReplicateMetadata แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , numReplicas แบบยาว, ตัวเลือก... ตัวเลือก)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ ReplicateMetadata ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
numReplicas จำนวนการจำลองการคำนวณ
ตัวเลือก มีค่าแอตทริบิวต์ทางเลือก
การส่งคืน
  • อินสแตนซ์ใหม่ของ ReplicateMetadata

สาธารณะคง ReplicateMetadata.Options อุปกรณ์การกำหนด (รายการ <ยาว> อุปกรณ์มอบหมาย)

พารามิเตอร์
การมอบหมายอุปกรณ์ การกำหนดอุปกรณ์สำหรับการคำนวณ

สาธารณะ ReplicateMetadata.Options hostComputeCore (รายการ <สตริง> hostComputeCore)

ReplicateMetadata.Options สาธารณะแบบคงที่ numCoresPerReplica (Long numCoresPerReplica)

พารามิเตอร์
numCoresPerReplica จำนวนคอร์ต่อเรพลิกา ใช้สำหรับโมเดลความเท่าเทียม

สาธารณะ ReplicateMetadata.Options paddingMap (รายการ <String> paddingMap)

สาธารณะ ReplicateMetadata.Options stepMarkerLocation แบบคงที่ (สตริง stepMarkerLocation)

โทโพโลยี ReplicateMetadata.Options สาธารณะ แบบคงที่ (โทโพโลยีสตริง)

พารามิเตอร์
โทโพโลยี TopologyProto ระบุโทโพโลยีของชิ้น TPU pod

ReplicateMetadata.Options แบบคงที่สาธารณะ useSpmdForXlaPartitioning (useSpmdForXlaPartitioning บูลีน)

สาธารณะ ReplicateMetadata.Options useTpu แบบคงที่ (useTpu บูลีน)

พารามิเตอร์
ใช้Tpu ไม่ว่าจะวางการคำนวณบน TPU