คลาสคงสาธารณะ 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 computationShape (รายการ <Long> computationShape)
พารามิเตอร์
รูปร่างการคำนวณ | เลิกใช้แล้ว ใช้ num_cores_per_replica แทน |
---|
สาธารณะ ReplicateMetadata.Options deviceAssignment (รายการ <Long> อุปกรณ์การกำหนด)
พารามิเตอร์
การมอบหมายอุปกรณ์ | การกำหนดอุปกรณ์สำหรับการคำนวณ |
---|
ReplicateMetadata สาธารณะ ตัวเลือก numCoresPerReplica (Long numCoresPerReplica)
พารามิเตอร์
numCoresPerReplica | จำนวนคอร์ต่อเรพลิกา ใช้สำหรับโมเดลความเท่าเทียม |
---|
โท โพโลยี ReplicateMetadata.Options สาธารณะ (โทโพโลยีสตริง)
พารามิเตอร์
โทโพโลยี | TopologyProto ระบุโทโพโลยีของชิ้น TPU pod |
---|
สาธารณะ ReplicateMetadata.Options useTpu (useTpu บูลีน)
พารามิเตอร์
ใช้Tpu | ไม่ว่าจะวางการคำนวณบน TPU |
---|