TPUReshardVariables

lớp cuối cùng công khai TPUReshardVariables

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
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.

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