警告:このAPIは非推奨であり、置き換えが安定した後 TensorFlowの将来のバージョンで削除される予定です。

このページは Cloud Translation API によって翻訳されました。
Switch to English

TPUReplicateMetadata

パブリックファイナルクラスTPUReplicateMetadata

TPU計算を複製する方法を示すメタデータ。

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

ネストされたクラス

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

パブリックメソッド

静的TPUReplicateMetadata.Options
allowSoftPlacement (ブール値allowSoftPlacement)
静的TPUReplicateMetadata.Options
computeShape (List <Long> computeShape)
静的TPUReplicateMetadata
createスコープスコープ、Long numReplicas、オプション...オプション)
新しいTPUReplicateMetadata操作をラップするクラスを作成するファクトリメソッド。
静的TPUReplicateMetadata.Options
deviceAssignment (List <Long> deviceAssignment)
静的TPUReplicateMetadata.Options
hostComputeCore (List <String> hostComputeCore)
静的TPUReplicateMetadata.Options
numCoresPerReplica (長いnumCoresPerReplica)
静的TPUReplicateMetadata.Options
paddingMap (List <String> paddingMap)
静的TPUReplicateMetadata.Options
stepMarkerLocation (文字列stepMarkerLocation)
静的TPUReplicateMetadata.Options
トポロジー(ストリングトポロジー)
静的TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (ブール値useSpmdForXlaPartitioning)
静的TPUReplicateMetadata.Options
useTpu (ブール値useTpu)

継承されたメソッド

パブリックメソッド

public static TPUReplicateMetadata.Options allowSoftPlacement (ブール値allowSoftPlacement)

パブリック静的TPUReplicateMetadata.Options computationShape(一覧<ロング> computationShape)

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

public static TPUReplicateMetadata create スコープスコープ、Long numReplicas、オプション...オプション)

新しいTPUReplicateMetadata操作をラップするクラスを作成するためのファクトリメソッド。

パラメーター
範囲現在のスコープ
numReplicas計算のレプリカの数
オプションオプションの属性値を運ぶ
戻り値
  • TPUReplicateMetadataの新しいインスタンス

public static TPUReplicateMetadata.Options deviceAssignment (List <Long> deviceAssignment)

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

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

public static TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

パラメーター
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 useSpmdForXlaPartitioning (ブール値useSpmdForXlaPartitioning)

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

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