คลาสสุดท้ายสาธารณะ SendTPUEmbeddingGradients
ดำเนินการอัปเดตการไล่ระดับสีของตารางที่ฝัง
ค่าคงที่
สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
SendTPUEmbeddingGradients แบบคงที่ |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
ค่าคงที่: "SendTPUEmbeddingGradients"
วิธีการสาธารณะ
สร้าง SendTPUEmbeddingGradients แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , Iterable < Operand < TFloat32 >> อินพุต, Iterable < Operand < TFloat32 >> LearningRates, การกำหนดค่าสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ SendTPUEmbeddingGradients ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
อินพุต | TensorList ของการไล่ระดับสีสำหรับอัปเดตตารางที่ฝัง อาร์กิวเมนต์นี้มีความยาวและรูปร่างเท่ากันกับค่าที่ส่งคืนของ RecvTPUEmbeddingActivations แต่มีการไล่ระดับสีของการสูญเสียของโมเดลที่เกี่ยวข้องกับการเปิดใช้งานการฝัง ตารางการฝังได้รับการอัปเดตจากการไล่ระดับสีเหล่านี้ผ่านเครื่องมือเพิ่มประสิทธิภาพที่ระบุในการกำหนดค่าการฝัง TPU ที่มอบให้กับ tpu.initialize_system |
อัตราการเรียนรู้ | TensorList ของสเกลาร์ float32 หนึ่งรายการสำหรับแท็กอัตราการเรียนรู้แบบไดนามิกแต่ละรายการ: ดูความคิดเห็นใน //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto หลายตารางสามารถแชร์แท็กอัตราการเรียนรู้แบบไดนามิกเดียวกันตามที่ระบุไว้ในการกำหนดค่า หากอัตราการเรียนรู้ของตารางทั้งหมดคงที่ รายการนี้ควรว่างเปล่า |
กำหนดค่า | โปรโตคอล TPUEmbeddingConfiguration แบบอนุกรม |
การส่งคืน
- อินสแตนซ์ใหม่ของ SendTPUEmbeddingGradients