공개 최종 클래스 EnqueueTPUEmbeddingIntegerBatch
입력 배치 텐서 목록을 TPUEmbedding에 추가하는 작업입니다.
중첩 클래스
| 수업 | EnqueueTPUEmbeddingIntegerBatch.Options | EnqueueTPUEmbeddingIntegerBatch 의 선택적 속성 | |
공개 방법
| 정적 EnqueueTPUEmbeddingIntegerBatch | |
| 정적 EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (긴 장치Ordinal) |
상속된 메서드
공개 방법
public static EnqueueTPUEmbeddingIntegerBatch create ( Scope 범위, Iterable< Operand <Integer>> 배치, Operand <String> modeOverride, Options... 옵션)
새로운 EnqueueTPUEmbeddingIntegerBatch 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
| 범위 | 현재 범위 |
|---|---|
| 일괄 | 테이블에 대한 인덱스를 포함하는 각 임베딩 테이블당 하나씩 1D 텐서 목록입니다. |
| 모드 재정의 | TPUEmbeddingConfiguration에 지정된 모드를 재정의하는 문자열 입력입니다. 지원되는 값은 {'unspecified', 'inference', 'training', 'backward_pass_only'}입니다. 'unspecified'로 설정되면 TPUEmbeddingConfiguration에 설정된 모드가 사용되고, 그렇지 않으면 mode_override가 사용됩니다. |
| 옵션 | 선택적 속성 값을 전달합니다. |
보고
- EnqueueTPUEmbeddingIntegerBatch의 새 인스턴스
공개 정적 EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
매개변수
| 장치서수 | 사용할 TPU 기기입니다. 0보다 크고 노드가 배치된 작업의 TPU 코어 수보다 작아야 합니다. |
|---|