op ที่ได้รับการเปิดใช้งานการฝังบน TPU
ระบบ TPU ดำเนินการค้นหาและการรวมการฝัง ผลลัพธ์ของการรวมเหล่านี้สามารถมองเห็นได้ในกราฟ Tensorflow เป็นผลลัพธ์ของ XlaRecvTPUEmbeddingActivations Op op นี้ส่งคืนรายการที่มีหนึ่ง Tensor ของการเปิดใช้งานต่อตารางที่ระบุในโมเดล
วิธีการสาธารณะ
คงที่ XlaRecvTPUEmbeddingActivations | |
Iterator < ตัวถูกดำเนินการ <Float>> | ตัววนซ้ำ () |
รายการ< เอาท์พุต <โฟลต>> | เอาท์พุท () TensorList ของการเปิดใช้งานการฝังที่มีหนึ่ง Tensor ต่อตารางการฝังในโมเดล |
วิธีการสืบทอด
วิธีการสาธารณะ
XlaRecvTPUEmbeddingActivations สาธารณะแบบคงที่ สร้าง ( ขอบเขต ขอบเขต, ตัวถูกดำเนินการ <?> deduplicationData, Long numTables , String config)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ XlaRecvTPUEmbeddingActivations ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
deduplicationData | Tensor with type=DT_VARIANT ที่มีข้อมูลการขจัดความซ้ำซ้อน เทนเซอร์คือทูเพิลที่ซ้อนกัน XLA ที่มีองค์ประกอบ N (โดยที่ N คืออัตราส่วนของจำนวนการฝังต่อเทนเซอร์คอร์ต่อชิป TPU) แต่ละองค์ประกอบของทูเพิลที่ซ้อนกันคือทูเพิลของเทนเซอร์อันดับ 1 เมตริกซ์แต่ละตัวมีดัชนี (DT_UIT32) สำหรับการฝังการค้นหาบน TensorCore หรือน้ำหนัก (DT_FLOAT) เพื่อนำไปใช้กับเอาต์พุตของการดำเนินการค้นหาแบบฝัง |
numTables | จำนวนเทนเซอร์การเปิดใช้งานเอาต์พุต หากมีตัวอธิบายคุณลักษณะในการกำหนดค่าการฝัง tpu จะเท่ากับจำนวนคุณลักษณะ มิฉะนั้นจะเท่ากับจำนวนตารางการฝังในโมเดล |
config | โปรโต TPUEmbeddingConfiguration แบบอนุกรม |
คืนสินค้า
- อินสแตนซ์ใหม่ของ XlaRecvTPUEmbeddingActivations
รายการสาธารณะ < เอาท์พุต <โฟลต>> เอาท์พุท ()
TensorList ของการเปิดใช้งานการฝังที่มีหนึ่ง Tensor ต่อตารางการฝังในโมเดล