NcclBroadcast

公共finalクラスNcclBroadcast

出力に接続されているすべてのデバイスに「入力」を送信します。

出力に接続されているすべてのデバイスに「入力」を送信します。

グラフは、出力に接続されているすべての操作に有効なデバイス割り当てがあり、操作自体にこれらのデバイスの1つが割り当てられるように作成する必要があります。

input:ブロードキャストへの入力。出力:入力と同じです。 shape:入力テンソルの形状。

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリックハンドルを返します。
静的<T拡張番号> NcclBroadcast <T>
作成スコープ範囲、オペランド<T>入力、図形の形状)
新しいNcclBroadcast操作をラップするクラスを作成するファクトリメソッド。
出力<T>
出力()

継承されたメソッド

パブリックメソッド

公共の出力<T> asOutput()

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

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

パブリック静的NcclBroadcast <T>を作成スコープ範囲、オペランド<T>入力、図形の形状)

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

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

公共の出力<T>出力()