Sebuah operasi yang mengeksekusi partisi TPUEmbedding pada konfigurasi pusat
perangkat dan menghitung ukuran HBM (dalam byte) yang diperlukan untuk operasi TPUEmbedding.
Metode Publik
Keluaran <String> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
Keluaran <String> | konfigurasi umum () Proto konfigurasi umum berkode string yang berisi metadata tentang output partisi TPUEmbedding dan ukuran HBM (dalam byte) yang diperlukan untuk pengoperasian. |
ExecuteTPUEmbeddingPartitioner statis |
Metode Warisan
Metode Publik
Keluaran publik <String> sebagaiOutput ()
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.
Keluaran publik <String> commonConfig ()
Proto konfigurasi umum berkode string yang berisi metadata tentang output partisi TPUEmbedding dan ukuran HBM (dalam byte) yang diperlukan untuk pengoperasian.
ExecuteTPUEmbeddingPartitioner statis publik buat ( Lingkup lingkup, Konfigurasi string)
Metode pabrik untuk membuat kelas yang membungkus operasi ExecuteTPUEmbeddingPartitioner baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
konfigurasi | Proto TPUEmbeddingConfiguration diserialkan ke string, menjelaskan konfigurasi TPUEmbedding yang diinginkan. |
Kembali
- contoh baru dari ExecuteTPUEmbeddingPartitioner