TPUReplicateMetadata

パブリック最終クラスTPUReplicateMetadata

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

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

ネストされたクラス

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

定数

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

パブリックメソッド

静的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
useSpmdForXlaPartitioning (ブール型 useSpmdForXlaPartitioning)
静的TPUReplicateMetadata.Options
useTpu (ブール型 useTpu)

継承されたメソッド

定数

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

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

定数値: "TPUReplicateMetadata"

パブリックメソッド

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 useSpmdForXlaPartitioning (ブール型 useSpmdForXlaPartitioning)

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

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