警告:このAPIは非推奨であり、置き換えが安定した後 TensorFlowの将来のバージョンで削除される予定です。

NcclAllReduce

コレクションでコンテンツを整理 必要に応じて、コンテンツの保存と分類を行います。
公共finalクラスNcclAllReduce

すべての入力テンソルにわたる縮小を含むテンソルを出力します。

同じ `shared_name内のopsに渡されたすべての入力テンソルにわたるリダクションを含むテンソルを出力します。

グラフは、1つの操作がshared_name値 `c`で実行される場合、` num_devices`操作がshared_name値 `c`で実行されるように作成する必要があります。そうしないと、グラフの実行が完了しなくなります。

input:削減データへの入力:すべての `num_devices`デバイスにわたる削減の値。削減:実行する削減操作。 num_devices:この削減に参加しているデバイスの数。 shared_name:同じ削減のops間で共有された識別子。

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリックハンドルを返します。
静的<T拡張番号> NcclAllReduce <T>
作成スコープ範囲、オペランド<T>入力、文字列削減、ロングnumDevices、ストリングsharedName)
新しいNcclAllReduce操作をラップするクラスを作成するファクトリメソッド。
出力<T>
データ()

継承されたメソッド

パブリックメソッド

公共の出力<T> asOutput()

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

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

パブリック静的NcclAllReduce <T>を作成スコープ範囲、オペランド<T>入力、文字列削減、ロングnumDevices、ストリングsharedName)

新しいNcclAllReduce操作をラップするクラスを作成するファクトリメソッド。

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

公共の出力<T>データ()