TPUReshardVariables

публичный финальный класс TPUReshardVariables

Операция, которая переводит переменные TPU на устройстве в указанное состояние.

Операция, которая переводит переменные TPU на устройстве в указанное состояние. Только для внутреннего пользования.

Состояние сегментирования представлено как ключ компиляции, которая сгенерировала программы сегментирования/расшардинга вместе с основной программой. new_format_key указывает желаемое состояние, а format_state_var — текущее состояние переменных.

Публичные методы

статические TPUReshardVariables

create ( Область действия, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)

Фабричный метод для создания класса, обертывающего новую операцию TPUReshardVariables.

Унаследованные методы

Из класса org.tensorflow.op.PrimitiveOp

последнее логическое значение	равно (Объект obj)
окончательный интервал	хэш-код ()
Операция	оп () Возвращает базовую `Operation`
последняя строка	нанизывать ()

Из класса java.lang.Object

логическое значение	равно (Объект arg0)
последний класс<?>	получитьКласс ()
интервал	хэш-код ()
окончательная пустота	поставить в известность ()
окончательная пустота	уведомитьВсе ()
Нить	нанизывать ()
окончательная пустота	подождать (длинный arg0, int arg1)
окончательная пустота	подождите (длинный arg0)
окончательная пустота	ждать ()

Публичные методы

public static TPUReshardVariables create (область действия , Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)

Фабричный метод для создания класса, обертывающего новую операцию TPUReshardVariables.

Параметры

объем	текущий объем

Возврат

новый экземпляр TPUReshardVariables