TPU実行と更新変数

パブリック最終クラスTPUExecuteAndUpdateVariables

オプションのインプレース変数更新を使用してプログラムを実行する Op。

(オプションで) デバイス変数を読み取り、TPU デバイス上で TPU プログラムをロードして実行し、次に (オプションで) 属性 device_var_reads_indices (変数を直接読み取るプログラム入力インデックス) および device_var_updates_indices で指定されているように、プログラム出力を使用して変数をインプレース更新します。 (変数の更新に使用されるプログラム出力インデックス、-1 は更新なし/読み取り専用を意味します)。このようなプログラム出力はこれらの変数によって消費され、OP 出力には表示されません。分散 TPU コンパイラの内部使用用。

パブリックメソッド

静的TPUExecuteAndUpdateVariables
create ( Scopeスコープ、Iterable< Operand <?>> args、 Operand <String> key、List<Class<?>> Tresults、List<Long> deviceVarReadsIndices、List<Long> deviceVarUpdatesIndices)
新しい TPUExecuteAndUpdateVariables オペレーションをラップするクラスを作成するファクトリ メソッド。
イテレータ<オペランド<オブジェクト>>
リスト<出力<?>>
結果()

継承されたメソッド

パブリックメソッド

public static TPUExecuteAndUpdateVariables create (スコープスコープ、Iterable< Operand <?>> args、 Operand <String> キー、List<Class<?>> Tresults、List<Long> deviceVarReadsIndices、List<Long> deviceVarUpdatesIndices)

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

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

public Iterator<オペランド<オブジェクト>> iterator ()

public List<出力<?>>結果()