TPUReshardVariables

공개 최종 클래스 TPUReshardVariables

기기 내 TPU 변수를 지정된 상태로 리샤딩하는 작업입니다.

기기 내 TPU 변수를 지정된 상태로 리샤딩하는 작업입니다. 내부 전용입니다.

샤딩 상태는 메인 프로그램과 함께 샤딩/샤딩 해제 프로그램을 생성한 컴파일의 키로 표시됩니다. new_format_key는 원하는 상태를 지정하고 format_state_var는 변수의 현재 상태입니다.

공개 방법

정적 TPUReshard변수
생성 ( Scope 범위, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)
새 TPUReshardVariables 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

상속된 메서드

공개 방법

public static TPUReshardVariables create ( Scope 범위, Iterable< Operand <?>> vars, Operand <String> newFormatKey, Operand <?> formatStateVar)

새 TPUReshardVariables 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

매개변수
범위 현재 범위
보고
  • TPUReshardVariables의 새 인스턴스