SendTPUEmbeddingGradients

คลาสสุดท้ายสาธารณะ SendTPUEmbeddingGradients

ดำเนินการอัปเดตการไล่ระดับสีของตารางที่ฝัง

ค่าคงที่

สตริง OP_NAME ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

วิธีการสาธารณะ

SendTPUEmbeddingGradients แบบคงที่
สร้าง ( ขอบเขตขอบเขต Iterable< Operand < TFloat32 >> อินพุต Iterable< Operand < TFloat32 >> LearningRates, การกำหนดค่าสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ 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