EnqueueTPUEmbeddingIntegerBatch

공개 최종 클래스 EnqueueTPUEmbeddingIntegerBatch

입력 배치 텐서 목록을 TPUEmbedding에 추가하는 작업입니다.

중첩 클래스

수업 EnqueueTPUEmbeddingIntegerBatch.Options EnqueueTPUEmbeddingIntegerBatch 의 선택적 속성

상수

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

공개 방법

정적 EnqueueTPUEmbeddingIntegerBatch
생성 ( Scope 범위, Iterable< Operand < TInt32 >> 배치, Operand < TString > modeOverride, 옵션... 옵션)
새로운 EnqueueTPUEmbeddingIntegerBatch 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
정적 EnqueueTPUEmbeddingIntegerBatch.Options
deviceOrdinal (긴 장치Ordinal)

상속된 메서드

상수

공개 정적 최종 문자열 OP_NAME

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

상수 값: "EnqueueTPUEmbeddingIntegerBatch"

공개 방법

public static EnqueueTPUEmbeddingIntegerBatch create ( Scope 범위, Iterable< Operand < TInt32 >> 배치, Operand < TString > modeOverride, 옵션... 옵션)

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

매개변수
범위 현재 범위
일괄 테이블에 대한 인덱스를 포함하는 각 임베딩 테이블당 하나씩 1D 텐서 목록입니다.
모드 재정의 TPUEmbeddingConfiguration에 지정된 모드를 재정의하는 문자열 입력입니다. 지원되는 값은 {'unspecified', 'inference', 'training', 'backward_pass_only'}입니다. 'unspecified'로 설정되면 TPUEmbeddingConfiguration에 설정된 모드가 사용되고, 그렇지 않으면 mode_override가 사용됩니다.
옵션 선택적 속성 값을 전달합니다.
보고
  • EnqueueTPUEmbeddingIntegerBatch의 새 인스턴스

공개 정적 EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)

매개변수
장치서수 사용할 TPU 기기입니다. 0보다 크고 노드가 배치된 작업의 TPU 코어 수보다 작아야 합니다.