ExecuteTPUEmbeddingPartitioner

คลาสสุดท้ายสาธารณะ ExecuteTPUmbeddingPartitioner

การดำเนินการที่เรียกใช้งานพาร์ติชัน TPUEmbedding บนการกำหนดค่าส่วนกลาง

อุปกรณ์และคำนวณขนาด HBM (เป็นไบต์) ที่จำเป็นสำหรับการดำเนินการ TPUEmbedding

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

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

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

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

สาธารณะ เอาท์พุท <String> asOutput ()

ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์

อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต

สาธารณะ เอาท์พุท <String> commonConfig ()

โปรโตคอลการกำหนดค่าทั่วไปที่เข้ารหัสสตริงประกอบด้วยข้อมูลเมตาเกี่ยวกับเอาต์พุตตัวแบ่งพาร์ติชัน TPUEmbedding และขนาด HBM (เป็นไบต์) ที่จำเป็นสำหรับการดำเนินการ

สร้าง ExecuteTPUEmbeddingPartitioner แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , การกำหนดค่าสตริง)

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
กำหนดค่า โปรโต TPUEmbeddingConfiguration ถูกทำให้เป็นอนุกรมเป็นสตริง โดยอธิบายการกำหนดค่า TPUEmbedding ที่ต้องการ
การส่งคืน
  • อินสแตนซ์ใหม่ของ ExecuteTPUEmbeddingPartitioner