Otomatik bölümlemeden otomatik bölümlemeye geçiş yapmak için XLA SPMD bölümleyici tarafından kullanılan bir işlem
manuel bölümleme. Girdiye (tam şekilli, otomatik olarak bölümlenecek) manuel bölümleme tarafından kullanılan aynı parçalamayla açıklama ekler ve daha sonra manuel olarak bölümlenen operasyonlar tarafından tüketilecek parça şeklinde bir tensör çıkışı sağlar. Şekil eşit şekilde bölümlenemiyorsa dolgu bölgesi 0'larla maskelenecektir.
Sabitler
Sicim | OP_NAME | Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı |
Genel Yöntemler
Çıkış <T> | Çıkış olarak () Tensörün sembolik tutamacını döndürür. |
statik <T, TType'ı genişletir > XlaSpmdFullToShardShape <T> | |
Çıkış <T> | çıktı () |
Kalıtsal Yöntemler
Sabitler
genel statik son Dize OP_NAME
Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı
Genel Yöntemler
genel Çıkış <T> asOutput ()
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 static XlaSpmdFullToShardShape <T> create ( Kapsam kapsamı , İşlenen <T> girişi, String manualSharding)
Yeni bir XlaSpmdFullToShardShape işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Parametreler
kapsam | mevcut kapsam |
---|
İadeler
- XlaSpmdFullToShardShape'in yeni bir örneği