TPUReplicateMetadata.Options

kelas statis publik TPUReplicationMetadata.Options

Atribut opsional untuk TPUReplicateMetadata

Metode Publik

TPUReplikatMetadata.Opsi
izinkanSoftPlacement (Boolean izinkanSoftPlacement)
TPUReplikatMetadata.Opsi
ComputationShape (Daftar<Panjang> ComputationShape)
TPUReplikatMetadata.Opsi
penugasan perangkat (Daftar<Panjang> penugasan perangkat)
TPUReplikatMetadata.Opsi
hostComputeCore (Daftar<String> hostComputeCore)
TPUReplikatMetadata.Opsi
numCoresPerReplica (NumCoresPerReplika panjang)
TPUReplikatMetadata.Opsi
paddingMap (Daftar<String> paddingMap)
TPUReplikatMetadata.Opsi
stepMarkerLocation (String stepMarkerLocation)
TPUReplikatMetadata.Opsi
topologi (Topologi string)
TPUReplikatMetadata.Opsi
useSpmdForXlaPartitioning (penggunaan BooleanSpmdForXlaPartitioning)
TPUReplikatMetadata.Opsi
useTpu (Boolean useTpu)

Metode Warisan

Metode Publik

TPUReplicationMetadata.Options publik memungkinkanSoftPlacement (Boolean memungkinkanSoftPlacement)

TPUReplicationMetadata.Options computingShape publik (Daftar<Panjang> ComputationShape)

Parameter
bentuk komputasi TIDAK DIGUNAKAN lagi. Gunakan num_cores_per_replica sebagai gantinya.

public TPUReplicationMetadata.Options deviceAssignment (Daftar<Panjang> deviceAssignment)

Parameter
penugasan perangkat Penugasan perangkat untuk komputasi.

publik TPUReplicationMetadata.Options hostComputeCore (Daftar<String> hostComputeCore)

TPUReplicationMetadata.Options publik numCoresPerReplica (NumCoresPerReplica panjang)

Parameter
numCoresPerReplika Jumlah inti per replika. Digunakan untuk paralelisme model.

publik TPUReplicationMetadata.Options paddingMap (Daftar<String> paddingMap)

publik TPUReplicationMetadata.Options stepMarkerLocation (String stepMarkerLocation)

topologi TPUReplicationMetadata.Options publik (Topologi string)

Parameter
topologi TopologyProto menunjukkan topologi potongan pod TPU.

TPUReplicationMetadata.Options publik useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

TPUReplikat Metadata publik. Opsi useTpu (Boolean useTpu)

Parameter
gunakanTpu Apakah akan menempatkan perhitungan pada TPU.