ExecuteTPUEmbeddingPartitioner

공개 최종 클래스 ExecuteTPUEmbeddingPartitioner

중앙 구성에서 TPUEmbedding 파티셔너를 실행하는 작업

TPUEmbedding 작업에 필요한 HBM 크기(바이트)를 계산합니다.

공개 방법

출력 <문자열>
출력 ()
텐서의 기호 핸들을 반환합니다.
출력 <문자열>
공통구성 ()
TPUEmbedding 파티셔너 출력 및 작업에 필요한 HBM 크기(바이트)에 대한 메타데이터가 포함된 문자열로 인코딩된 일반 구성 proto입니다.
정적 ExecuteTPUEmbeddingPartitioner
생성 ( 범위 범위, 문자열 구성)
새로운 ExecuteTPUEmbeddingPartitioner 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

상속된 메서드

공개 방법

공개 출력 <String> asOutput ()

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

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

공개 출력 <String> commonConfig ()

TPUEmbedding 파티셔너 출력 및 작업에 필요한 HBM 크기(바이트)에 대한 메타데이터가 포함된 문자열로 인코딩된 일반 구성 proto입니다.

공개 정적 ExecuteTPUEmbeddingPartitioner 생성 ( 범위 범위, 문자열 구성)

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

매개변수
범위 현재 범위
구성 원하는 TPUEmbedding 구성을 설명하는 문자열로 직렬화된 TPUEmbeddingConfiguration 프로토콜입니다.
보고
  • ExecuteTPUEmbeddingPartitioner의 새 인스턴스