TPUReshardVariables

TPUReshardVariables kelas akhir publik

Op yang mengatur ulang variabel TPU pada perangkat ke status yang ditentukan.

Op yang mengatur ulang variabel TPU pada perangkat ke status yang ditentukan. Hanya penggunaan internal.

Status sharding direpresentasikan sebagai kunci kompilasi yang menghasilkan program sharding/unsharding bersama dengan program utama. new_format_key menentukan keadaan yang diinginkan, dan format_state_var adalah keadaan variabel saat ini.

Metode Publik

Variabel TPUReshard statis
buat ( Lingkup lingkup, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUReshardVariables baru.

Metode Warisan

Metode Publik

TPUReshardVariables statis publik dibuat ( Lingkup cakupan, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)

Metode pabrik untuk membuat kelas yang membungkus operasi TPUReshardVariables baru.

Parameter
cakupan ruang lingkup saat ini
Kembali
  • contoh baru dari TPUReshardVariables