ExecuteTPUEmbeddingPartitioner

genel final sınıfı ExecuteTPUEmbeddingPartitioner

Merkezi konfigürasyonda TPUEmbedding bölümleyicisini çalıştıran bir işlem

cihaza bağlanır ve TPU Yerleştirme işlemi için gereken HBM boyutunu (bayt cinsinden) hesaplar.

Genel Yöntemler

Çıkış <Dize>
Çıkış olarak ()
Bir tensörün sembolik tutamacını döndürür.
Çıkış <Dize>
ortak Yapılandırma ()
TPUEmbedding bölümleyici çıktısı ve işlem için gereken HBM boyutu (bayt cinsinden) hakkında meta veriler içeren dize kodlu bir ortak yapılandırma protokolü.
statik ExecuteTPUEmbeddingPartitioner
oluştur ( Kapsam kapsamı, Dize yapılandırması)
Yeni bir ExecuteTPUEmbeddingPartitioner işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.

Kalıtsal Yöntemler

Genel Yöntemler

public Output <String> 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.

public Output <String> commonConfig ()

TPUEmbedding bölümleyici çıktısı ve işlem için gereken HBM boyutu (bayt cinsinden) hakkında meta veriler içeren dize kodlu bir ortak yapılandırma protokolü.

public static ExecuteTPUEmbeddingPartitioner create ( Kapsam kapsamı, Dize yapılandırması)

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

Parametreler
kapsam mevcut kapsam
yapılandırma İstenilen TPUEmbedding yapılandırmasını açıklayan bir dizeye serileştirilmiş bir TPUEmbeddingConfiguration protokolü.
İadeler
  • ExecuteTPUEmbeddingPartitioner'ın yeni bir örneği