공개 최종 클래스 TPUReshardVariables
기기 내 TPU 변수를 지정된 상태로 리샤딩하는 작업입니다.
기기 내 TPU 변수를 지정된 상태로 리샤딩하는 작업입니다. 내부 전용입니다.
샤딩 상태는 메인 프로그램과 함께 샤딩/샤딩 해제 프로그램을 생성한 컴파일의 키로 표시됩니다. new_format_key는 원하는 상태를 지정하고 format_state_var는 변수의 현재 상태입니다.
공개 방법
정적 TPUReshard변수 |
상속된 메서드
공개 방법
public static TPUReshardVariables create ( Scope 범위, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
새 TPUReshardVariables 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|
보고
- TPUReshardVariables의 새 인스턴스