TPUOrdinalSelector

パブリック最終クラスTPUOrdinalSelector

TPUコアセレクターOp.

この Op は、TPU プログラムを実行するための TPU コアのセット (ウォームアップ用) または単一の TPU コア (通常の推論用) を生成します。出力は TPUPartitionedCall によって消費されます。

パブリックメソッド

出力<整数>
asOutput ()
テンソルのシンボリック ハンドルを返します。
静的TPUOrdinalSelector
create (スコープスコープ)
新しい TPUOrdinalSelector オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<整数>
デバイス序数()
ベクトル 1 つ以上の TPU コア。

継承されたメソッド

パブリックメソッド

public Output <整数> asOutput ()

テンソルのシンボリック ハンドルを返します。

TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。

public static TPUOrdinalSelector create (スコープスコープ)

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

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

public Output <Integer> deviceOrdinals ()

ベクトル 1 つ以上の TPU コア。