Op sẽ phân chia lại các biến TPU trên thiết bị về trạng thái được chỉ định.
Op sẽ phân chia lại các biến TPU trên thiết bị về trạng thái được chỉ định. Chỉ sử dụng nội bộ.
Trạng thái sharding được biểu diễn dưới dạng khóa của quá trình biên dịch tạo ra các chương trình sharding/unsharding cùng với chương trình chính. new_format_key chỉ định trạng thái mong muốn và format_state_var là trạng thái hiện tại của các biến.
Phương pháp công khai
TPUReshardCác biến tĩnh |
Phương pháp kế thừa
Phương pháp công khai
public static TPUReshardVariables tạo ( Phạm vi phạm vi, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TPUReshardVariables mới.
Thông số
phạm vi | phạm vi hiện tại |
---|
Trả lại
- một phiên bản mới của TPUReshardVariables