공개 최종 클래스 ExecuteTPUEmbeddingPartitioner
중앙 구성에서 TPUEmbedding 파티셔너를 실행하는 작업
TPUEmbedding 작업에 필요한 HBM 크기(바이트)를 계산합니다.
공개 방법
출력 <문자열> | 출력 () 텐서의 기호 핸들을 반환합니다. |
출력 <문자열> | 공통구성 () TPUEmbedding 파티셔너 출력 및 작업에 필요한 HBM 크기(바이트)에 대한 메타데이터가 포함된 문자열로 인코딩된 일반 구성 proto입니다. |
정적 ExecuteTPUEmbeddingPartitioner |
상속된 메서드
공개 방법
공개 출력 <String> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
공개 출력 <String> commonConfig ()
TPUEmbedding 파티셔너 출력 및 작업에 필요한 HBM 크기(바이트)에 대한 메타데이터가 포함된 문자열로 인코딩된 일반 구성 proto입니다.
공개 정적 ExecuteTPUEmbeddingPartitioner 생성 ( 범위 범위, 문자열 구성)
새로운 ExecuteTPUEmbeddingPartitioner 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
구성 | 원하는 TPUEmbedding 구성을 설명하는 문자열로 직렬화된 TPUEmbeddingConfiguration 프로토콜입니다. |
보고
- ExecuteTPUEmbeddingPartitioner의 새 인스턴스