パブリック静的クラスInfeedEnqueueTuple.Options
InfeedEnqueueTuple
のオプションの属性
パブリックメソッド
InfeedEnqueueTuple.Options | deviceOrdinal (長い deviceOrdinal) |
InfeedEnqueueTuple.Options | レイアウト(List<Long> レイアウト) |
継承されたメソッド
パブリックメソッド
public InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)
パラメーター
デバイス通常 | 使用する TPU デバイス。これは、Op が TPU デバイスで実行されている場合は -1 である必要があり、Op が CPU デバイスで実行されている場合は >= 0 である必要があります。 |
---|
public InfeedEnqueueTuple.Optionsレイアウト(List<Long> レイアウト)
パラメーター
レイアウト | すべてのタプル形状について、要求されたレイアウトをマイナーからメジャーの順序で保持するベクトル (形状が「形状」入力に表示される順序)。サブシェイプのレイアウト要素は -1 に設定できます。この場合、対応するレイアウトはインフィード操作によって計算されます。 |
---|