TPUReshardVariables

パブリック最終クラスTPUReshardVariables

デバイス上の TPU 変数を指定された状態にリシャードする Op。

デバイス上の TPU 変数を指定された状態にリシャードする Op。内部でのみ使用。

シャーディング状態は、メイン プログラムとともにシャーディング/アンシャーディング プログラムを生成したコンパイルのキーとして表されます。 new_format_key は目的の状態を指定し、format_state_var は変数の現在の状態です。

パブリックメソッド

静的TPUReshardVariables
create (スコープscope、Iterable< Operand <?>> vars、 Operand <String> newFormatKey、 Operand <?> formatStateVar)
新しい TPUReshardVariables オペレーションをラップするクラスを作成するためのファクトリ メソッド。

継承されたメソッド

パブリックメソッド

public static TPUReshardVariables create (スコープscope、Iterable< Operand <?>> vars、 Operand <String> newFormatKey、 Operand <?> formatStateVar)

新しい TPUReshardVariables オペレーションをラップするクラスを作成するためのファクトリ メソッド。

パラメーター
範囲現在のスコープ
戻り値
  • TPUReshardVariables の新しいインスタンス