kelas statis publik TPUReplicateMetadata.Options
Atribut opsional untuk TPUReplicateMetadata
Metode Publik
TPUReplicateMetadata.Options | allowSoftPlacement (Boolean allowSoftPlacement) |
TPUReplicateMetadata.Options | komputasiShape (Daftar<Long> komputasiShape) |
TPUReplicateMetadata.Options | deviceAssignment (Daftar<Long> deviceAssignment) |
TPUReplicateMetadata.Options | hostComputeCore (Daftar<String> hostComputeCore) |
TPUReplicateMetadata.Options | numCoresPerReplica (Jumlah panjangCoresPerReplica) |
TPUReplicateMetadata.Options | paddingMap (Daftar<String> paddingMap) |
TPUReplicateMetadata.Options | stepMarkerLocation (String stepMarkerLocation) |
TPUReplicateMetadata.Options | topologi (topologi string) |
TPUReplicateMetadata.Options | tpuCompileOptionsProto (String tpuCompileOptionsProto) |
TPUReplicateMetadata.Options | gunakanSpmdForXlaPartitioning (penggunaan BooleanSpmdForXlaPartitioning) |
TPUReplicateMetadata.Options | useTpu (Boolean useTpu) |
Metode yang Diwarisi
Metode Publik
public TPReplicateMetadata.Options computingShape ( List<Long> computingShape)
Parameter
komputasiBentuk | TIDAK DIGUNAKAN LAGI. Gunakan num_cores_per_replica sebagai gantinya. |
---|
publik TPUReplicateMetadata.Options deviceAssignment (Daftar<Long> deviceAssignment)
Parameter
tugas perangkat | Penugasan perangkat untuk perhitungan. |
---|
TPUReplicateMetadata.Options publik numCoresPerReplica (Long numCoresPerReplica)
Parameter
numCoresPerReplica | Jumlah inti per replika. Digunakan untuk model paralelisme. |
---|
topologi TPUReplicateMetadata.Options publik (topologi String)
Parameter
topologi | TopologyProto menunjukkan topologi dari potongan pod TPU. |
---|
TPUReplicateMetadata.Options publik menggunakanSpmdForXlaPartitioning (penggunaan BooleanSpmdForXlaPartitioning)
publik TPUReplicateMetadata.Options useTpu (Boolean useTpu)
Parameter
gunakanTpu | Apakah akan menempatkan perhitungan pada TPU. |
---|