TPUReplicateMetadata

パブリック最終クラスTPUReplicateMetadata

TPU 計算をどのように複製するかを示すメタデータ。

この操作は、「tpu.replicate()」計算サブグラフの操作に共通のメタデータを保持します。

ネストされたクラス

クラスTPUReplicateMetadata.Options TPUReplicateMetadataのオプションの属性

パブリックメソッド

静的TPUReplicateMetadata.Options
allowSoftPlacement (ブール値のallowSoftPlacement)
静的TPUReplicateMetadata.Options
computationShape (List<Long> computationShape)
静的TPUReplicateMetadata
create (スコープscope、Long numReplicas、オプション...オプション)
新しい 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.OptionsallowSoftPlacement (ブール値のallowSoftPlacement)

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 hostComputeCore (List<String> hostComputeCore)

public static TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

パラメーター
レプリカあたりのコア数レプリカごとのコアの数。モデルの並列処理に使用されます。

public static TPUReplicateMetadata.Options paddingMap (List<String> paddingMap)

public static TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

public static TPUReplicateMetadata.Optionsトポロジ(文字列トポロジ)

パラメーター
トポロジーTPU ポッド スライスのトポロジを示す TopologyProto。

public static TPUReplicateMetadata.Options tpuCompileOptionsProto (String tpuCompileOptionsProto)

public static TPUReplicateMetadata.Options useSpmdForXlaPartitioning (ブール型 useSpmdForXlaPartitioning)

public static TPUReplicateMetadata.Options useTpu (ブール型 useTpu)

パラメーター
Tpuを使用する計算を TPU に配置するかどうか。