ปฏิบัติการที่ได้รับการเปิดใช้งานการฝังบน 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