EnqueueTPUEmbeddingIntegerBatch

パブリック最終クラスEnqueueTPUEmbeddingIntegerBatch

入力バッチ テンソルのリストを TPUEmbedding にエンキューする操作。

ネストされたクラス

クラスEnqueueTPUEmbeddingIntegerBatch.Options EnqueueTPUEmbeddingIntegerBatchのオプションの属性

パブリックメソッド

静的EnqueueTPUEmbeddingIntegerBatch
create (スコープスコープ、Iterable<オペランド<Integer>> バッチ、オペランド<String> modeOverride、オプション...オプション)
新しい EnqueueTPUEmbeddingIntegerBatch オペレーションをラップするクラスを作成するファクトリ メソッド。
静的EnqueueTPUEmbeddingIntegerBatch.Options
deviceOrdinal (長い deviceOrdinal)

継承されたメソッド

パブリックメソッド

public static EnqueueTPUEmbeddingIntegerBatch create (スコープスコープ、Iterable<オペランド<Integer>> バッチ、オペランド<String> modeOverride、オプション...オプション)

新しい EnqueueTPUEmbeddingIntegerBatch オペレーションをラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
バッチ1D テンソルのリスト。埋め込みテーブルごとに 1 つあり、テーブルへのインデックスが含まれます。
モードオーバーライドTPUEmbeddingConfiguration で指定されたモードをオーバーライドする文字列入力。サポートされている値は、{'unspecified'、'inference'、'training'、'backward_pass_only'} です。 「unspecified」に設定すると、TPUEmbeddingConfiguration で設定されたモードが使用され、それ以外の場合は、mode_override が使用されます。
オプションオプションの属性値を持ちます
戻り値
  • EnqueueTPUEmbeddingIntegerBatch の新しいインスタンス

public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)

パラメーター
デバイス通常使用する TPU デバイス。 0 以上で、ノードが配置されているタスク内の TPU コアの数未満である必要があります。