TPURoundRobin

genel final sınıfı TPURoundRobin

TPU çekirdeklerinde döngüsel yük dengeleme.

TPU çekirdekleri arasında dönüşümlü olarak çalışan bir yük dengeleme işlemi.

Bu işlem, [0, NumTPUCoresVisiblePerHost] içindeki tamsayılar arasında döner. 'TPUPPartitionedCall' gibi hesaplamaların yürütüleceği bir TPU çekirdeğini girdi olarak alan TensorFlow işlemleriyle arayüz oluşturmak için kullanışlıdır.

Device_ordinal: [0, NumTPUCoresVisiblePerHost] cinsinden bir tamsayı.

Genel Yöntemler

Çıkış <Tamsayı>
Çıkış olarak ()
Bir tensörün sembolik tutamacını döndürür.
Statik TPURoundRobin
oluştur ( Kapsam kapsamı)
Yeni bir TPURoundRobin işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Çıkış <Tamsayı>

Kalıtsal Yöntemler

Genel Yöntemler

public Çıkış <Tamsayı> asOutput ()

Bir tensörün sembolik tutamacını döndürür.

TensorFlow işlemlerinin girdileri, başka bir TensorFlow işleminin çıktılarıdır. Bu yöntem, girişin hesaplanmasını temsil eden sembolik bir tanıtıcı elde etmek için kullanılır.

genel statik TPURoundRobin oluşturma ( Kapsam kapsamı)

Yeni bir TPURoundRobin işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.

Parametreler
kapsam mevcut kapsam
İadeler
  • TPURoundRobin'in yeni bir örneği

public Çıkış <Tamsayı> DeviceOrdinal ()