ExecuteTPUEmbeddingPartitioner

パブリック最終クラスExecuteTPUEmbeddingPartitioner

中央構成で TPUEmbedding パーティショナーを実行する操作

デバイスを検索し、TPUEmbedding 操作に必要な HBM サイズ (バイト単位) を計算します。

パブリックメソッド

出力<文字列>
asOutput ()
テンソルのシンボリック ハンドルを返します。
出力<文字列>
commonConfig ()
TPUEmbedding パーティショナー出力に関するメタデータと、操作に必要な HBM サイズ (バイト単位) を含む文字列エンコードされた共通構成プロトコル。
静的ExecuteTPUEmbeddingPartitioner
create (スコープスコープ、文字列構成)
新しい ExecuteTPUEmbeddingPartitioner オペレーションをラップするクラスを作成するファクトリ メソッド。

継承されたメソッド

パブリックメソッド

public Output <String> asOutput ()

テンソルのシンボリック ハンドルを返します。

TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。

public出力<String> commonConfig ()

TPUEmbedding パーティショナー出力に関するメタデータと、操作に必要な HBM サイズ (バイト単位) を含む文字列エンコードされた共通構成プロトコル。

public static ExecuteTPUEmbeddingPartitioner create (スコープスコープ、文字列構成)

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

パラメーター
範囲現在のスコープ
構成文字列にシリアル化された TPUEmbeddingConfiguration プロト。必要な TPUEmbedding 構成を記述します。
戻り値
  • ExecuteTPUEmbeddingPartitioner の新しいインスタンス