EnqueueTPUEmbeddingIntegerBatch

パブリック最終クラスEnqueueTPUEmbeddingIntegerBatch

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

ネストされたクラス

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

定数

OP_NAME TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

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

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: "EnqueueTPUEmbeddingIntegerBatch"

パブリックメソッド

public static EnqueueTPUEmbeddingIntegerBatch create (スコープスコープ、Iterable< Operand < TInt32 >> バッチ、 Operand < TString > modeOverride、 Options...オプション)

新しい 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 コアの数未満である必要があります。