パブリック最終クラスReplicateMetadata
TPU 計算をどのように複製するかを示すメタデータ。
この操作は、「tpu.replicate()」計算サブグラフの操作に共通のメタデータを保持します。
ネストされたクラス
| クラス | ReplicateMetadata.Options | ReplicateMetadataのオプションの属性 | |
定数
| 弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
| 静的ReplicateMetadata.Options | allowSoftPlacement (ブール値のallowSoftPlacement) |
| 静的ReplicateMetadata.Options | computationShape (List<Long> computationShape) |
| 静的なReplicateMetadata | |
| 静的ReplicateMetadata.Options | deviceAssignment (List<Long> deviceAssignment) |
| 静的ReplicateMetadata.Options | hostComputeCore (List<String> hostComputeCore) |
| 静的ReplicateMetadata.Options | numCoresPerReplica (長い numCoresPerReplica) |
| 静的ReplicateMetadata.Options | パディングマップ(List<String> パディングマップ) |
| 静的ReplicateMetadata.Options | stepMarkerLocation (文字列 stepMarkerLocation) |
| 静的ReplicateMetadata.Options | トポロジ(文字列トポロジ) |
| 静的ReplicateMetadata.Options | useSpmdForXlaPartitioning (ブール型 useSpmdForXlaPartitioning) |
| 静的ReplicateMetadata.Options | useTpu (ブール型 useTpu) |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
定数値: "TPUReplicateMetadata"
パブリックメソッド
public static ReplicateMetadata.Options computationShape (List<Long> computationShape)
パラメーター
| 計算形状 | 非推奨。代わりに num_cores_per_replica を使用してください。 |
|---|
public static ReplicateMetadata create (スコープscope、Long numReplicas、 Options...オプション)
新しい ReplicateMetadata オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| レプリカ数 | 計算のレプリカの数 |
| オプション | オプションの属性値を持ちます |
戻り値
- ReplicateMetadata の新しいインスタンス
public static ReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
パラメーター
| デバイスの割り当て | 計算のためのデバイスの割り当て。 |
|---|
public static ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
パラメーター
| レプリカあたりのコア数 | レプリカごとのコアの数。モデルの並列処理に使用されます。 |
|---|
public static ReplicateMetadata.Optionsトポロジ(String トポロジ)
パラメーター
| トポロジー | TPU ポッド スライスのトポロジを示す TopologyProto。 |
|---|