ReplicateMetadata

genel son sınıf ReplicateMetadata

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 ReplicateMetadata.Options ReplicateMetadata için isteğe bağlı özellikler

Sabitler

Sicim OP_NAME Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı

Genel Yöntemler

statik ReplicateMetadata.Options
AllowSoftPlacement (BooleanallowSoftPlacement)
statik ReplicateMetadata.Options
computationShape (Liste<Uzun> computationShape)
statik ReplicateMetadata
oluştur ( Kapsam kapsamı, Uzun sayıReplikalar, Seçenekler... seçenekler)
Yeni bir ReplicateMetadata işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
statik ReplicateMetadata.Options
cihazAtaması (Liste<Uzun> cihazAtaması)
statik ReplicateMetadata.Options
hostComputeCore (Liste<String> hostComputeCore)
statik ReplicateMetadata.Options
numCoresPerReplica (Uzun numCoresPerReplica)
statik ReplicateMetadata.Options
paddingMap (Liste<String> paddingMap)
statik ReplicateMetadata.Options
stepMarkerLocation (Dize stepMarkerLocation)
statik ReplicateMetadata.Options
topoloji (Dize topolojisi)
statik ReplicateMetadata.Options
useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
statik ReplicateMetadata.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 ReplicateMetadata.Options AllowSoftPlacement (BooleanallowSoftPlacement)

public static ReplicateMetadata.Options computationShape (List<Long> computationShape)

Parametreler
hesaplamaŞekil KULLANIMDAN KALDIRILDI. Bunun yerine num_cores_per_replica kullanın.

public static ReplicateMetadata oluştur ( Kapsam kapsamı, Uzun numReplicas, Seçenekler... seçenekler)

Yeni bir ReplicateMetadata 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
  • ReplicateMetadata'nın yeni bir örneği

public static ReplicateMetadata.Options DeviceAssignment (List<Long> DeviceAssignment)

Parametreler
cihazAtaması Hesaplama için cihazların atanması.

public static ReplicateMetadata.Options hostComputeCore (List<String> hostComputeCore)

public static ReplicateMetadata.Options numCoresPerReplica (Uzun numCoresPerReplica)

Parametreler
numCoresPerReplica Çoğaltma başına çekirdek sayısı. Model paralelliği için kullanılır.

public static ReplicateMetadata.Options paddingMap (List<String> paddingMap)

public static ReplicateMetadata.Options stepMarkerLocation (String stepMarkerLocation)

public static ReplicateMetadata.Options topolojisi (Dize topolojisi)

Parametreler
topoloji TopologyProto, TPU bölme diliminin topolojisini belirtir.

public static ReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)

public static ReplicateMetadata.Options useTpu (Boolean useTpu)

Parametreler
Tpu'yu kullan Hesaplamanın TPU'ya yerleştirilip yerleştirilmeyeceği.