XlaSpmdFullToShardShape

공개 최종 클래스 XlaSpmdFullToShardShape

XLA SPMD 파티셔너가 자동 파티셔닝에서 다음 파티셔닝으로 전환하는 데 사용하는 작업입니다.

수동 파티셔닝. 수동 분할에 사용되는 것과 동일한 샤딩으로 입력(전체 형태, 자동으로 분할됨)에 주석을 달고 나중에 수동으로 분할된 작업에서 사용할 샤드 모양 텐서를 출력합니다. 모양이 균등하게 분할 가능하지 않은 경우 패딩 영역은 0으로 마스크됩니다.

상수

OP_NAME TensorFlow 코어 엔진에서 알려진 이 작업의 이름

공개 방법

출력 <T>
출력 ()
텐서의 기호 핸들을 반환합니다.
static <T는 TType을 확장합니다. > XlaSpmdFullToShardShape <T>
생성 ( 범위 범위, 피연산자 <T> 입력, 문자열 manualSharding)
새로운 XlaSpmdFullToShardShape 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <T>
출력 ()

상속된 메서드

상수

공개 정적 최종 문자열 OP_NAME

TensorFlow 코어 엔진에서 알려진 이 작업의 이름

상수 값: "XlaSpmdFullToShardShape"

공개 방법

공개 출력 <T> asOutput ()

텐서의 기호 핸들을 반환합니다.

TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.

public static XlaSpmdFullToShardShape <T> create ( 범위 범위, 피연산자 <T> 입력, 문자열 manualSharding)

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

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

공개 출력 <T> 출력 ()