เชื่อมต่ออินพุต N กับการคำนวณ TPU ที่จำลองแบบ N-way
การดำเนินการนี้เก็บอินพุตที่จำลองแบบไว้ในกราฟย่อยการคำนวณ `tpu.replicate()` อินพุตที่ถูกจำลองแต่ละรายการจะมีรูปร่างและประเภทเหมือนกันควบคู่ไปกับเอาต์พุต
ตัวอย่างเช่น:
%a = "tf.opA"()
%b = "tf.opB"()
%replicated_input = "tf.TPUReplicatedInput"(%a, %b)
%computation = "tf.Computation"(%replicated_input)
การคำนวณข้างต้นมีอินพุตที่จำลองแบบของสองแบบจำลอง คลาสที่ซ้อนกัน
ระดับ | TPUReplicatedInput ตัวเลือก | แอตทริบิวต์ทางเลือกสำหรับ TPUReplicatedInput |
ค่าคงที่
สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
เอาท์พุต <T> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
คงที่ <T ขยาย TType > TPUReplicatedInput <T> | สร้าง (ขอบเขต ขอบเขต , อินพุต < Operand <T>>, ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUReplicatedInput ใหม่ |
TPUReplicatedInput.Options แบบคงที่ | ดัชนี (ดัชนียาว) |
TPUReplicatedInput.Options แบบคงที่ | isMirroredVariable (บูลีน isMirroredVariable) |
TPUReplicatedInput.Options แบบคงที่ | isPacked (บูลีน isPacked) |
เอาท์พุต <T> | เอาท์พุท () |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
วิธีการสาธารณะ
เอาท์พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สร้าง TPUReplicatedInput <T> แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , อินพุต < ตัวดำเนินการ <T>> ตัวเลือก, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUReplicatedInput ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- อินสแตนซ์ใหม่ของ TPUReplicatedInput