パブリック最終クラスInfeedEnqueueTuple
複数の Tensor 値を XLA タプルとして計算にフィードします。
ネストされたクラス
クラス | InfeedEnqueueTuple.Options | InfeedEnqueueTuple のオプションの属性 |
パブリックメソッド
静的InfeedEnqueueTuple | create ( Scopeスコープ、Iterable< Operand <?>> 入力、List< Shape > シェイプ、 Options...オプション) 新しい InfeedEnqueueTuple オペレーションをラップするクラスを作成するファクトリ メソッド。 |
静的InfeedEnqueueTuple.Options | deviceOrdinal (長い deviceOrdinal) |
静的InfeedEnqueueTuple.Options | レイアウト(List<Long> レイアウト) |
継承されたメソッド
パブリックメソッド
public static InfeedEnqueueTuple create (スコープスコープ、Iterable< Operand <?>> 入力、List< Shape > シェイプ、 Options...オプション)
新しい InfeedEnqueueTuple オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
入力 | インフィードメカニズムを使用して提供されるテンソルのリスト。 |
形 | 「入力」の各テンソルの形状。 |
オプション | オプションの属性値を持ちます |
戻り値
- InfeedEnqueueTuple の新しいインスタンス
public static InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)
パラメーター
デバイス通常 | 使用する TPU デバイス。これは、Op が TPU デバイスで実行されている場合は -1 である必要があり、Op が CPU デバイスで実行されている場合は >= 0 である必要があります。 |
---|
public static InfeedEnqueueTuple.Optionsレイアウト(List<Long> レイアウト)
パラメーター
レイアウト | すべてのタプル形状について、要求されたレイアウトをマイナーからメジャーの順序で保持するベクトル (形状が「形状」入力に表示される順序)。サブシェイプのレイアウト要素は -1 に設定できます。この場合、対応するレイアウトはインフィード操作によって計算されます。 |
---|