Операция, которая переводит переменные TPU на устройстве в указанное состояние.
Операция, которая переводит переменные TPU на устройстве в указанное состояние. Только для внутреннего пользования.
Состояние сегментирования представлено как ключ компиляции, которая сгенерировала программы сегментирования/расшардинга вместе с основной программой. new_format_key указывает желаемое состояние, а format_state_var — текущее состояние переменных.
Публичные методы
статические TPUReshardVariables |
Унаследованные методы
Публичные методы
public static TPUReshardVariables create (область действия , Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
Фабричный метод для создания класса, обертывающего новую операцию TPUReshardVariables.
Параметры
объем | текущий объем |
---|
Возврат
- новый экземпляр TPUReshardVariables