Op, która ponownie przenosi zmienne TPU na urządzeniu do określonego stanu.
Op, która ponownie przenosi zmienne TPU na urządzeniu do określonego stanu. Do użytku wewnętrznego.
Stan fragmentowania jest reprezentowany jako klucz kompilacji, która wygenerowała programy dzielenia/odłamywania wraz z programem głównym. new_format_key określa pożądany stan, a format_state_var to bieżący stan zmiennych.
Metody publiczne
statyczne zmienne TPUReshard |
Metody dziedziczone
Metody publiczne
public static TPUReshardVariables create ( zakres zakresu , Iterowalny < Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUReshardVariables.
Parametry
zakres | aktualny zakres |
---|
Zwroty
- nowa instancja TPUReshardVariables