คำเตือน: API นี้เลิกใช้งานแล้วและจะถูกลบออกใน TensorFlow เวอร์ชันในอนาคตหลังจาก การแทนที่ เสถียรแล้ว

ExecuteTPUEmbeddingPartitioner

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

op ที่รันตัวแบ่งพาร์ติชัน TPUEmbedding บนการกำหนดค่าส่วนกลาง

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

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

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

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

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

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

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

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

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

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

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

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

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