パブリック最終クラスTPUReshardVariables
デバイス上の TPU 変数を指定された状態にリシャードする Op。
デバイス上の TPU 変数を指定された状態にリシャードする Op。内部でのみ使用。
シャーディング状態は、メイン プログラムとともにシャーディング/アンシャーディング プログラムを生成したコンパイルのキーとして表されます。 new_format_key は目的の状態を指定し、format_state_var は変数の現在の状態です。
パブリックメソッド
静的TPUReshardVariables |
継承されたメソッド
パブリックメソッド
public static TPUReshardVariables create (スコープscope、Iterable< Operand <?>> vars、 Operand <String> newFormatKey、 Operand <?> formatStateVar)
新しい TPUReshardVariables オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|
戻り値
- TPUReshardVariables の新しいインスタンス