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 |
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