ปฏิบัติการที่ได้รับการเปิดใช้งานการฝังบน TPU
ระบบ TPU ดำเนินการค้นหาแบบฝังและการรวมกลุ่มที่ระบุโดยอาร์กิวเมนต์ของ TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch ผลลัพธ์ของการรวมกลุ่มเหล่านี้จะมองเห็นได้ในกราฟ Tensorflow เป็นเอาต์พุตของ RecvTPUEmbeddingActivations op op นี้ส่งคืนรายการที่มีหนึ่งเทนเซอร์ของการเปิดใช้งานต่อตารางที่ระบุในโมเดล สามารถมี RecvTPUEmbeddingActivations ได้สูงสุด 1 รายการในกราฟ TPU
ค่าคงที่
| สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
| RecvTPUEmbeddingActivations แบบคงที่ | |
| ตัววนซ้ำ < ตัวดำเนินการ < TFloat32 >> | ตัววนซ้ำ () |
| รายการ < เอาท์พุต < TFloat32 >> | เอาท์พุท () TensorList ของการเปิดใช้งานการฝังที่มีหนึ่ง Tensor ต่อตารางการฝังในโมเดล |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
วิธีการสาธารณะ
สร้าง RecvTPUEmbeddingActivations สาธารณะแบบคงที่ (ขอบเขต ขอบเขต , numOutputs แบบยาว, การกำหนดค่าสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ RecvTPUEmbeddingActivations ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| numOutputs | จำนวนเทนเซอร์การเปิดใช้งานเอาต์พุต เท่ากับจำนวนตารางที่ฝังในโมเดล |
| กำหนดค่า | โปรโตคอล TPUEmbeddingConfiguration แบบอนุกรม |
การส่งคืน
- อินสแตนซ์ใหม่ของ RecvTPUEmbeddingActivations