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