TPUReshardVariables

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

ปฏิบัติการที่แบ่งตัวแปร TPU บนอุปกรณ์ใหม่เป็นสถานะที่ระบุ

ปฏิบัติการที่แบ่งตัวแปร TPU บนอุปกรณ์ใหม่เป็นสถานะที่ระบุ ใช้ภายในเท่านั้น.

สถานะการแบ่งส่วนจะแสดงเป็นคีย์ของการคอมไพล์ที่สร้างโปรแกรมการแบ่งส่วน/การแยกส่วนพร้อมกับโปรแกรมหลัก new_format_key ระบุสถานะที่ต้องการ และ format_state_var คือสถานะปัจจุบันของตัวแปร

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

TPUReshardVariables แบบคงที่
สร้าง (ขอบเขต ขอบเขต Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUReshardVariables ใหม่

วิธีการสืบทอด

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

สร้าง TPUReshardVariables แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUReshardVariables ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
การส่งคืน
  • อินสแตนซ์ใหม่ของ TPUReshardVariables