ReplicateMetadata

パブリック最終クラスReplicateMetadata

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

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

ネストされたクラス

クラス

ReplicateMetadata.Options

ReplicateMetadataのオプションの属性

定数

弦

OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

静的ReplicateMetadata.Options	allowSoftPlacement (ブール値のallowSoftPlacement)
静的ReplicateMetadata.Options	computationShape (List<Long> computationShape)
静的なReplicateMetadata	create (スコープscope、Long numReplicas、オプション...オプション) 新しい ReplicateMetadata オペレーションをラップするクラスを作成するファクトリ メソッド。
静的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)

継承されたメソッド

クラスorg.tensorflow.op.RawOpから

最終ブール値	等しい(オブジェクトオブジェクト)
最終整数	ハッシュコード（）
手術	オプ() この計算単位を単一のOperationとして返します。
最後の文字列	toString ()

クラスjava.lang.Objectから

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

インターフェースorg.tensorflow.op.Opから

抽象的な実行環境	環境() このオペレーションが作成された実行環境を返します。
抽象的な操作	オプ() この計算単位を単一のOperationとして返します。

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: "TPUReplicateMetadata"

パブリックメソッド

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

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

パラメーター

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

public static ReplicateMetadata create (スコープscope、Long numReplicas、 Options...オプション)

新しい ReplicateMetadata オペレーションをラップするクラスを作成するファクトリ メソッド。

パラメーター

範囲	現在のスコープ
レプリカ数	計算のレプリカの数
オプション	オプションの属性値を持ちます

戻り値

• ReplicateMetadata の新しいインスタンス

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

パラメーター

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

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

public static ReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

パラメーター

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

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

public static ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

public static ReplicateMetadata.Optionsトポロジ(String トポロジ)

パラメーター

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

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

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

パラメーター

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