InfeedEnqueueTuple

パブリック最終クラス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 に設定できます。この場合、対応するレイアウトはインフィード操作によって計算されます。