Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

ExecuteTPUEmbeddingPartitioner

kelas akhir publik ExecuteTPUEmbeddingPartitioner

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
buat ( Lingkup lingkup, konfigurasi String)
Metode pabrik untuk membuat kelas yang membungkus operasi ExecuteTPUEmbbeddingPartitioner baru.

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