ReplicateMetadata.Options

คลาสคงสาธารณะ ReplicateMetadata.Options

แอ็ตทริบิวต์ทางเลือกสำหรับ ReplicateMetadata

วิธีการสาธารณะ

ReplicateMetadata.Options
AllowSoftPlacement (บูลีน AllowSoftPlacement)
ReplicateMetadata.Options
computationShape (รายการ <Long> computationShape)
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)

วิธีการสืบทอด

วิธีการสาธารณะ

ReplicateMetadata.Options สาธารณะ อนุญาตให้ SoftPlacement (บูลีน AllowSoftPlacement)

สาธารณะ ReplicateMetadata.Options computationShape (รายการ <Long> computationShape)

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

สาธารณะ ReplicateMetadata.Options deviceAssignment (รายการ <Long> อุปกรณ์การกำหนด)

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

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

ReplicateMetadata สาธารณะ ตัวเลือก 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