TPUReplicateMetadata

kelas akhir publik TPUReplicateMetadata

Metadata yang menunjukkan bagaimana penghitungan TPU harus direplikasi.

Operasi ini menyimpan metadata yang umum untuk operasi subgraf komputasi `tpu.replicate()`.

Kelas Bersarang

kelas

TPUReplicateMetadata.Options

Atribut opsional untuk TPUReplicateMetadata

Metode Publik

TPUReplicateMetadata.Options statis	allowSoftPlacement (Boolean allowSoftPlacement)
TPUReplicateMetadata.Options statis	komputasiShape (Daftar<Long> komputasiShape)
TPUReplicateMetadata statis	buat ( Lingkup lingkup, Jumlah panjang Replika, Opsi ... opsi) Metode pabrik untuk membuat kelas yang membungkus operasi TPReplicateMetadata baru.
TPUReplicateMetadata.Options statis	deviceAssignment (Daftar<Long> deviceAssignment)
TPUReplicateMetadata.Options statis	hostComputeCore (Daftar<String> hostComputeCore)
TPUReplicateMetadata.Options statis	numCoresPerReplica (Jumlah panjangCoresPerReplica)
TPUReplicateMetadata.Options statis	paddingMap (Daftar<String> paddingMap)
TPUReplicateMetadata.Options statis	stepMarkerLocation (String stepMarkerLocation)
TPUReplicateMetadata.Options statis	topologi (topologi string)
TPUReplicateMetadata.Options statis	tpuCompileOptionsProto (String tpuCompileOptionsProto)
TPUReplicateMetadata.Options statis	gunakanSpmdForXlaPartitioning (penggunaan BooleanSpmdForXlaPartitioning)
TPUReplicateMetadata.Options statis	useTpu (Boolean useTpu)

Metode yang Diwarisi

Dari kelas org.tensorflow.op.PrimitiveOp

boolean akhir	sama (objek objek)
akhir	kode hash ()
Operasi	op () Mengembalikan Operation yang mendasarinya
string terakhir	toString ()

Dari kelas java.lang.Object

boolean	sama dengan (Objek arg0)
Kelas terakhir<?>	dapatkanKelas ()
ke dalam	kode hash ()
kekosongan akhir	beri tahu ()
kekosongan akhir	beri tahu Semua ()
Rangkaian	toString ()
kekosongan akhir	tunggu (arg panjang0, int arg1)
kekosongan akhir	tunggu (arg panjang0)
kekosongan akhir	tunggu ()

Metode Publik

publik TPReplicateMetadata.Options allowSoftPlacement (Boolean allowSoftPlacement)

public static TPReplicateMetadata.Options computingShape ( List<Long> computingShape)

Parameter

komputasiBentuk	TIDAK DIGUNAKAN LAGI. Gunakan num_cores_per_replica sebagai gantinya.

pembuatan TPReplicateMetadata statis publik ( Cakupan cakupan, Jumlah panjang, Opsi ... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi TPReplicateMetadata baru.

Parameter

cakupan	lingkup saat ini
jumlahReplika	Jumlah replika perhitungan
pilihan	membawa nilai atribut opsional

Kembali

• contoh baru TPUReplicateMetadata

TPUReplicateMetadata.Options deviceAssignment statis publik (List<Long> deviceAssignment)

Parameter

tugas perangkat	Penugasan perangkat untuk perhitungan.

publik statis TPUReplicateMetadata.Options hostComputeCore (Daftar<String> hostComputeCore)

TPUReplicateMetadata.Options publik statis numCoresPerReplica ( NumCoresPerReplica panjang)

Parameter

numCoresPerReplica	Jumlah inti per replika. Digunakan untuk model paralelisme.

publik TPReplicateMetadata.Options paddingMap (List<String> paddingMap)

publik TPReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

topologi TPUReplicateMetadata.Options publik statis (topologi String)

Parameter

topologi	TopologyProto menunjukkan topologi dari potongan pod TPU.

publik statis TPUReplicateMetadata.Options tpuCompileOptionsProto (String tpuCompileOptionsProto)

TPUReplicateMetadata.Options publik statis menggunakanSpmdForXlaPartitioning (penggunaan BooleanSpmdForXlaPartitioning)

TPUReplicateMetadata.Options public static useTpu (Boolean useTpu)

Parameter

gunakanTpu	Apakah akan menempatkan perhitungan pada TPU.