genel final sınıfı TPUReplicateMetadata
TPU hesaplamasının nasıl çoğaltılması gerektiğini gösteren meta veriler.
Bu işlem, bir "tpu.replicate()" hesaplama alt grafiğinin işlemleri için ortak olan meta verileri tutar.
İç İçe Sınıflar
| sınıf | TPUReplicateMetadata.Options | TPUReplicateMetadata için isteğe bağlı özellikler | |
Genel Yöntemler
| statik TPUReplicateMetadata.Options | AllowSoftPlacement (BooleanallowSoftPlacement) |
| statik TPUReplicateMetadata.Options | computationShape (Liste<Uzun> computationShape) |
| statik TPUÇoğaltılmış Meta Veriler | oluştur ( Kapsam kapsamı, Uzun sayıReplikalar, Seçenekler... seçenekler) Yeni bir TPUReplicateMetadata işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi. |
| statik TPUReplicateMetadata.Options | cihazAtaması (Liste<Uzun> cihazAtaması) |
| statik TPUReplicateMetadata.Options | hostComputeCore (Liste<String> hostComputeCore) |
| statik TPUReplicateMetadata.Options | numCoresPerReplica (Uzun numCoresPerReplica) |
| statik TPUReplicateMetadata.Options | paddingMap (Liste<String> paddingMap) |
| statik TPUReplicateMetadata.Options | stepMarkerLocation (Dize stepMarkerLocation) |
| statik TPUReplicateMetadata.Options | topoloji (Dize topolojisi) |
| statik TPUReplicateMetadata.Options | useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning) |
| statik TPUReplicateMetadata.Options | useTpu (Boolean useTpu) |
Kalıtsal Yöntemler
Genel Yöntemler
public static TPUReplicateMetadata.Options computationShape (List<Long> computationShape)
Parametreler
| hesaplamaŞekil | KULLANIMDAN KALDIRILDI. Bunun yerine num_cores_per_replica kullanın. |
|---|
public static TPUReplicateMetadata oluştur ( Kapsam kapsamı, Uzun numReplicas, Seçenekler... seçenekler)
Yeni bir TPUReplicateMetadata işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Parametreler
| kapsam | mevcut kapsam |
|---|---|
| numReplicas | Hesaplamanın kopya sayısı |
| seçenekler | isteğe bağlı nitelik değerlerini taşır |
İadeler
- TPUReplicateMetadata'nın yeni bir örneği
public static TPUReplicateMetadata.Options DeviceAssignment (List<Long> DeviceAssignment)
Parametreler
| cihazAtaması | Hesaplama için cihazların atanması. |
|---|
public static TPUReplicateMetadata.Options numCoresPerReplica (Uzun numCoresPerReplica)
Parametreler
| numCoresPerReplica | Çoğaltma başına çekirdek sayısı. Model paralelliği için kullanılır. |
|---|
public static TPUReplicateMetadata.Options topolojisi (Dize topolojisi)
Parametreler
| topoloji | TopologyProto, TPU bölme diliminin topolojisini belirtir. |
|---|
genel statik TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
genel statik TPUReplicateMetadata.Options useTpu (Boolean useTpu)
Parametreler
| Tpu'yu kullan | Hesaplamanın TPU'ya yerleştirilip yerleştirilmeyeceği. |
|---|