Sebuah op yang mengeksekusi partisi TPUembedding pada konfigurasi pusat
perangkat dan menghitung ukuran HBM (dalam byte) yang diperlukan untuk operasi penyisipan TPUE.
Metode Publik
Keluaran <String> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
Keluaran <String> | konfigurasi umum () Proto konfigurasi umum yang disandikan string yang berisi metadata tentang keluaran pemartisi TPUEbedding dan ukuran HBM (dalam byte) yang diperlukan untuk operasi. |
static ExecuteTPUEmbeddingPartitioner |
Metode yang Diwarisi
Metode Publik
Output publik <String> sebagai Output ()
Mengembalikan pegangan simbolis tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
Output publik <String> commonConfig ()
Proto konfigurasi umum yang disandikan string yang berisi metadata tentang keluaran pemartisi TPUEbedding dan ukuran HBM (dalam byte) yang diperlukan untuk operasi.
Buat ExecuteTPUEmbbeddingPartitioner statis publik ( Cakupan lingkup, Konfigurasi string)
Metode pabrik untuk membuat kelas yang membungkus operasi ExecuteTPUEmbbeddingPartitioner baru.
Parameter
cakupan | lingkup saat ini |
---|---|
konfigurasi | Proto TPUEmbeddingConfiguration diserialkan ke string, menjelaskan konfigurasi TPUEbedding yang diinginkan. |
Kembali
- contoh baru dari ExecuteTPUEmbeddingPartitioner