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 |
Sabitler
Sicim | OP_NAME | Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı |
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
Sabitler
genel statik son Dize OP_NAME
Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı
Sabit Değer: "TPUReplicateMetadata"
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. |
---|