TPUReplicateMetadata

คลาสสุดท้าย TPUReplicateMetadata

ข้อมูลเมตาที่ระบุว่าควรจำลองการคำนวณ TPU อย่างไร

การดำเนินการนี้เก็บข้อมูลเมตาร่วมกับการดำเนินการของกราฟย่อยการคำนวณ `tpu.replicate()`

คลาสที่ซ้อนกัน

ระดับ

TPUReplicateMetadata.Options

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

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

คงที่ TPUReplicateMetadata.Options	allowSoftPlacement (บูลีน allowSoftPlacement)
คงที่ TPUReplicateMetadata.Options	การคำนวณรูปร่าง (รายการ<ยาว> การคำนวณรูปร่าง)
TPUReplicateMetadata แบบคงที่	สร้าง ( ขอบเขต ขอบเขต, numReplicas แบบยาว, ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ TPUReplicateMetadata ใหม่
คงที่ TPUReplicateMetadata.Options	deviceAssignment (รายการ<Long> การกำหนดอุปกรณ์)
คงที่ TPUReplicateMetadata.Options	hostComputeCore (รายการ <String> hostComputeCore)
คงที่ TPUReplicateMetadata.Options	numCoresPerReplica (ยาว numCoresPerReplica)
คงที่ TPUReplicateMetadata.Options	paddingMap (รายการ <String> paddingMap)
คงที่ TPUReplicateMetadata.Options	stepMarkerLocation (สตริง stepMarkerLocation)
คงที่ TPUReplicateMetadata.Options	โทโพโลยี (โทโพโลยีสตริง)
คงที่ TPUReplicateMetadata.Options	tpuCompileOptionsProto (สตริง tpuCompileOptionsProto)
คงที่ TPUReplicateMetadata.Options	useSpmdForXlaPartitioning (ใช้บูลีน SpmdForXlaPartitioning)
คงที่ TPUReplicateMetadata.Options	useTpu (ใช้บูลีนTpu)

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

จากคลาส org.tensorflow.op.PrimitiveOp

บูลีนสุดท้าย	เท่ากับ (วัตถุ obj)
intสุดท้าย	แฮชโค้ด ()
การดำเนินการ	ออฟ () ส่งกลับ Operation พื้นฐาน
สตริงสุดท้าย	toString ()

จากคลาส java.lang.Object

บูลีน	เท่ากับ (วัตถุ arg0)
คลาสสุดท้าย<?>	getClass ()
int	แฮชโค้ด ()
สุดท้ายเป็นโมฆะ	แจ้ง ()
สุดท้ายเป็นโมฆะ	แจ้งทั้งหมด ()
สตริง	toString ()
สุดท้ายเป็นโมฆะ	รอ (arg ยาว, int arg1)
สุดท้ายเป็นโมฆะ	รอ (ยาว arg0)
สุดท้ายเป็นโมฆะ	รอ ()

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

TPUReplicateMetadata แบบคงที่สาธารณะ ตัวเลือก allowSoftPlacement (บูลีน allowSoftPlacement)

สาธารณะ TPUReplicateMetadata.Options การคำนวณ รูปร่าง (รายการ<ยาว> การคำนวณรูปร่าง)

พารามิเตอร์

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

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

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

พารามิเตอร์

ขอบเขต	ขอบเขตปัจจุบัน
numReplicas	จำนวนการจำลองของการคำนวณ
ตัวเลือก	ดำเนินการค่าแอตทริบิวต์ทางเลือก

คืนสินค้า

• อินสแตนซ์ใหม่ของ TPUReplicateMetadata

TPUReplicateMetadata.Options deviceAssignment สาธารณะ แบบคงที่ (List<Long> deviceAssignment)

พารามิเตอร์

อุปกรณ์Assignment	การกำหนดอุปกรณ์สำหรับการคำนวณ

TPUReplicateMetadata.Options สาธารณะแบบคง ที่ hostComputeCore (รายการ <String> hostComputeCore)

สาธารณะ TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

พารามิเตอร์

numCoresPerReplica	จำนวนคอร์ต่อเรพลิกา ใช้สำหรับแบบจำลองขนาน

TPUReplicateMetadata.Options paddingMap แบบคงที่สาธารณะ ( รายการ <String> paddingMap)

สาธารณะ TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

โทโพโลยี ตัวเลือก TPUReplicateMetadata.Options แบบคงที่สาธารณะ (โทโพโลยีสตริง)

พารามิเตอร์

โทโพโลยี	TopologyProto ระบุโทโพโลยีของชิ้นส่วน TPU pod

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

TPUReplicateMetadata แบบคงที่สาธารณะ ตัวเลือก useSpmdForXlaPartitioning (ใช้บูลีนSpmdForXlaPartitioning)

สาธารณะ TPUReplicateMetadata.Options useTpu (บูลีน useTpu)

พารามิเตอร์

ใช้Tpu	จะวางการคำนวณบน TPU หรือไม่