รับข้อมูลการขจัดความซ้ำซ้อน (ดัชนีและน้ำหนัก) จากคอร์ที่ฝัง
ข้อมูลการขจัดความซ้ำซ้อนคือ Tensor ที่มี type=DT_VARIANT เทนเซอร์นั้นเป็นทูเพิลที่ซ้อนกันของ XLA ซึ่งมีองค์ประกอบ N (โดยที่ N คืออัตราส่วนของจำนวนการฝังต่อเทนเซอร์คอร์ต่อชิป TPU) แต่ละองค์ประกอบของทูเพิลที่ซ้อนกันคือทูเพิลของเทนเซอร์อันดับ 1 เทนเซอร์แต่ละตัวมีดัชนี (DT_UINT32) สำหรับการค้นหาแบบฝังบน TensorCore หรือตุ้มน้ำหนัก (DT_FLOAT) เพื่อใช้กับเอาต์พุตของการดำเนินการค้นหาแบบฝัง
วิธีการสาธารณะ
เอาต์พุต <วัตถุ> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
XlaRecvTPUEmbeddingDeduplicationData แบบคงที่ | |
เอาท์พุต <?> | เอาท์พุท () |
วิธีการสืบทอด
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <Object> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สร้าง XlaRecvTPUEmbeddingDeduplicationData สาธารณะแบบคงที่ (ขอบเขต ขอบเขต , การกำหนดค่าสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ XlaRecvTPUEmbeddingDeduplicationData ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
กำหนดค่า | โปรโตคอล TPUEmbeddingConfiguration แบบอนุกรม |
การส่งคืน
- อินสแตนซ์ใหม่ของ XlaRecvTPUEmbeddingDeduplicationData