パブリック最終クラスEnqueueTPUEmbeddingIntegerBatch
入力バッチ テンソルのリストを TPUEmbedding にエンキューする操作。
ネストされたクラス
クラス | EnqueueTPUEmbeddingIntegerBatch.Options | EnqueueTPUEmbeddingIntegerBatch のオプションの属性 |
パブリックメソッド
静的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 コアの数未満である必要があります。 |
---|