ExecuteTPUEmbeddingPartitioner

kelas akhir publik ExecuteTPUEmbeddingPartitioner

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

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