Op yang mengubah variabel TPU pada perangkat ke status yang ditentukan.
Op yang mengubah variabel TPU pada perangkat ke status yang ditentukan. Penggunaan internal saja.
Status sharding direpresentasikan sebagai kunci kompilasi yang menghasilkan program sharding/unsharding bersama dengan program utama. new_format_key menentukan status yang diinginkan, dan format_state_var adalah status variabel saat ini.
Metode Publik
statis TPUReshardVariables |
Metode yang Diwarisi
Metode Publik
public static TPUReshardVariables membuat ( Scope lingkup, Iterable < Operan <? >> vars, Operan <String> newFormatKey, Operan <?> formatStateVar)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUReshardVariables baru.
Parameter
cakupan | lingkup saat ini |
---|
Kembali
- contoh baru TPUReshardVariables