ReplicateMetadata.Options

パブリック静的クラスReplicateMetadata.Options

ReplicateMetadataのオプションの属性

パブリックメソッド

ReplicateMetadata.Options
allowSoftPlacement (ブール値のallowSoftPlacement)
ReplicateMetadata.Options
computationShape (List<Long> computationShape)
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)

継承されたメソッド

パブリックメソッド

public ReplicateMetadata.OptionsallowSoftPlacement (ブール値のallowSoftPlacement)

public ReplicateMetadata.Options computationShape (List<Long> computationShape)

パラメーター
計算形状非推奨。代わりに num_cores_per_replica を使用してください。

public ReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)

パラメーター
デバイスの割り当て計算のためのデバイスの割り当て。

public ReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

public ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

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

public ReplicateMetadata.Options paddingMap (List<String> paddingMap)

public ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

パブリックReplicateMetadata.Optionsトポロジ(文字列トポロジ)

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

public ReplicateMetadata.Options useSpmdForXlaPartitioning (ブール型 useSpmdForXlaPartitioning)

public ReplicateMetadata.Options useTpu (ブール型 useTpu)

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