パブリック最終クラス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。 |
---|