คลาสสุดท้ายสาธารณะ ExecuteTPUmbeddingPartitioner
การดำเนินการที่เรียกใช้งานพาร์ติชัน TPUEmbedding บนการกำหนดค่าส่วนกลาง
อุปกรณ์และคำนวณขนาด HBM (เป็นไบต์) ที่จำเป็นสำหรับการดำเนินการ TPUEmbedding
วิธีการสาธารณะ
| เอาท์พุต <สตริง> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
| เอาท์พุต <สตริง> | การกำหนดค่าทั่วไป () โปรโตคอลการกำหนดค่าทั่วไปที่เข้ารหัสสตริงประกอบด้วยข้อมูลเมตาเกี่ยวกับเอาต์พุตตัวแบ่งพาร์ติชัน TPUEmbedding และขนาด HBM (เป็นไบต์) ที่จำเป็นสำหรับการดำเนินการ |
| ExecuteTPUEmbeddingPartitioner แบบคงที่ |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะ เอาท์พุท <String> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สาธารณะ เอาท์พุท <String> commonConfig ()
โปรโตคอลการกำหนดค่าทั่วไปที่เข้ารหัสสตริงประกอบด้วยข้อมูลเมตาเกี่ยวกับเอาต์พุตตัวแบ่งพาร์ติชัน TPUEmbedding และขนาด HBM (เป็นไบต์) ที่จำเป็นสำหรับการดำเนินการ
สร้าง ExecuteTPUEmbeddingPartitioner แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , การกำหนดค่าสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ ExecuteTPUEmbeddingPartitioner ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| กำหนดค่า | โปรโต TPUEmbeddingConfiguration ถูกทำให้เป็นอนุกรมเป็นสตริง โดยอธิบายการกำหนดค่า TPUEmbedding ที่ต้องการ |
การส่งคืน
- อินสแตนซ์ใหม่ของ ExecuteTPUEmbeddingPartitioner