XlaSpmdFullToShardShape

パブリック最終クラスXlaSpmdFullToShardShape

XLA SPMD パーティショナーによって自動パーティショニングから

手動パーティショニング。これは、手動パーティショニングで使用されるのと同じシャーディングで入力 (自動的にパーティショニングされるフルシェイプ) にアノテーションを付け、後で手動でパーティショニングされる操作で消費されるシャード形状のテンソルを出力します。形状が均等に分割できない場合、パディング領域は 0 でマスクされます。

定数

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

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
static <T extends TType > XlaSpmdFullToShardShape <T>
create (スコープscope、オペランド<T>入力、文字列manualSharding)
新しい XlaSpmdFullToShardShape オペレーションをラップするクラスを作成するファクトリ メソッド。
出力<T>

継承されたメソッド

定数

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

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

定数値: "XlaSpmdFullToShardShape"

パブリックメソッド

public Output <T> asOutput ()

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

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

public static XlaSpmdFullToShardShape <T> create (スコープscope、オペランド<T>入力、文字列manualSharding)

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

パラメーター
範囲現在のスコープ
戻り値
  • XlaSpmdFullToShardShape の新しいインスタンス

public出力<T>出力()