TPUReplicateMetadata.Options

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

TPUReplicateMetadataのオプションの属性

パブリックメソッド

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

継承されたメソッド

クラスjava.lang.Objectから

ブール値	等しい(オブジェクト arg0)
最終クラス<?>	getクラス()
整数	ハッシュコード（）
最後の空白	通知する()
最後の空白	すべて通知()
弦	toString ()
最後の空白	wait (long arg0, int arg1)
最後の空白	待機(長い引数0)
最後の空白	待って（）

パブリックメソッド

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

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

パラメーター

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

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

パラメーター

デバイスの割り当て	計算のためのデバイスの割り当て。

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

public TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

パラメーター

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

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

public TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

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

パラメーター

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

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

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

パラメーター

Tpuを使用する	計算を TPU に配置するかどうか。