パブリック最終クラスExecuteTPUEmbeddingPartitioner
中央構成で TPUEmbedding パーティショナーを実行する操作
デバイスを検索し、TPUEmbedding 操作に必要な HBM サイズ (バイト単位) を計算します。
パブリックメソッド
| 出力<文字列> | asOutput () テンソルのシンボリック ハンドルを返します。 |
| 出力<文字列> | commonConfig () TPUEmbedding パーティショナー出力に関するメタデータと、操作に必要な HBM サイズ (バイト単位) を含む文字列エンコードされた共通構成プロトコル。 |
| 静的ExecuteTPUEmbeddingPartitioner |
継承されたメソッド
パブリックメソッド
public Output <String> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public出力<String> commonConfig ()
TPUEmbedding パーティショナー出力に関するメタデータと、操作に必要な HBM サイズ (バイト単位) を含む文字列エンコードされた共通構成プロトコル。
public static ExecuteTPUEmbeddingPartitioner create (スコープスコープ、文字列構成)
新しい ExecuteTPUEmbeddingPartitioner オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| 構成 | 文字列にシリアル化された TPUEmbeddingConfiguration プロト。必要な TPUEmbedding 構成を記述します。 |
戻り値
- ExecuteTPUEmbeddingPartitioner の新しいインスタンス