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