TPUReplicateMetadata

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 AllowSoftPlacement (BooleanallowSoftPlacement)

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 hostComputeCore (List<String> hostComputeCore)

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 paddingMap (List<String> paddingMap)

public static TPUReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

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.