パブリック最終クラスXlaSpmdFullToShardShape
XLA SPMD パーティショナーによって自動パーティショニングから
手動パーティショニング。これは、手動パーティショニングで使用されるのと同じシャーディングで入力 (自動的にパーティショニングされるフルシェイプ) にアノテーションを付け、後で手動でパーティショニングされる操作で消費されるシャード形状のテンソルを出力します。形状が均等に分割できない場合、パディング領域は 0 でマスクされます。
定数
弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static <T extends TType > XlaSpmdFullToShardShape <T> | |
出力<T> | 出力() |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
定数値: "XlaSpmdFullToShardShape"
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static XlaSpmdFullToShardShape <T> create (スコープscope、オペランド<T>入力、文字列manualSharding)
新しい XlaSpmdFullToShardShape オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|
戻り値
- XlaSpmdFullToShardShape の新しいインスタンス