ปฏิบัติการที่แบ่งตัวแปร TPU บนอุปกรณ์ใหม่เป็นสถานะที่ระบุ
ปฏิบัติการที่แบ่งตัวแปร TPU บนอุปกรณ์ใหม่เป็นสถานะที่ระบุ ใช้ภายในเท่านั้น.
สถานะการแบ่งส่วนจะแสดงเป็นคีย์ของการคอมไพล์ที่สร้างโปรแกรมการแบ่งส่วน/การแยกส่วนพร้อมกับโปรแกรมหลัก new_format_key ระบุสถานะที่ต้องการ และ format_state_var คือสถานะปัจจุบันของตัวแปร
วิธีการสาธารณะ
TPUReshardVariables แบบคงที่ |
วิธีการสืบทอด
วิธีการสาธารณะ
สร้าง TPUReshardVariables แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUReshardVariables ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|
การส่งคืน
- อินสแตนซ์ใหม่ของ TPUReshardVariables