TPUReshardVariables

publiczna klasa końcowa TPUReshardVariables

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
utwórz (zakres zakresu , Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUReshardVariables.

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