TPURoundRobin

パブリック最終クラスTPURoundRobin

TPU コア上のラウンドロビン負荷分散。

TPU コア間でラウンドロビンを行う負荷分散処理。

この操作は、[0, NumTPUCoresVisiblePerHost] の整数間でラウンドロビンを実行します。これは、「TPUPartitionedCall」など、計算を実行する TPU コアを入力として受け取る TensorFlow 演算とのインターフェースに役立ちます。

device_ordinal: [0, NumTPUCoresVisiblePerHost] の整数。

パブリックメソッド

出力<整数>
asOutput ()
テンソルのシンボリック ハンドルを返します。
静的TPURoundRobin
create (スコープスコープ)
新しい TPURoundRobin オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<整数>
deviceOrdinal ()

継承されたメソッド

パブリックメソッド

public Output <整数> asOutput ()

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

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

public static TPURoundRobin create (スコープscope)

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

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

public Output <整数> deviceOrdinal ()