Sebuah operasi yang digunakan oleh partisi XLA SPMD untuk beralih dari partisi otomatis ke
partisi manual. Ini memberi anotasi pada input (bentuk penuh, untuk dipartisi secara otomatis) dengan sharding yang sama yang digunakan oleh partisi manual, dan menghasilkan tensor berbentuk shard untuk digunakan oleh operasi yang kemudian dipartisi secara manual. Jika bentuknya tidak dapat dipartisi secara merata, wilayah padding akan ditutupi dengan 0s.
Konstanta
Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
statis <T memperluas TType > XlaSpmdFullToShardShape <T> | |
Keluaran <T> | keluaran () |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
Keluaran publik <T> sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static XlaSpmdFullToShardShape <T> buat ( Lingkup lingkup, masukan Operan <T>, String manualSharding)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaSpmdFullToShardShape baru.
Parameter
cakupan | ruang lingkup saat ini |
---|
Kembali
- contoh baru XlaSpmdFullToShardShape