OrdinalSelector

공개 최종 클래스 OrdinalSelector

TPU 코어 선택기 Op.

이 작업은 TPU 프로그램을 실행할 TPU 코어 세트(워밍업용) 또는 단일 TPU 코어(일반 추론용)를 생성합니다. 출력은 TPUPartitionedCall에서 사용됩니다.

상수

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

공개 방법

출력 < TInt32 >
출력 ()
텐서의 기호 핸들을 반환합니다.
정적 서수 선택기
생성 ( 범위 범위)
새로운 OrdinalSelector 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 < TInt32 >
장치서수 ()
벡터는 1개 이상의 TPU 코어입니다.

상속된 메서드

상수

공개 정적 최종 문자열 OP_NAME

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

상수 값: "TPUOrdinalSelector"

공개 방법

공개 출력 < TInt32 > asOutput ()

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

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

공개 정적 OrdinalSelector 작성 ( 범위 범위)

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

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

공개 출력 <TInt32> deviceOrdinals ()

벡터는 1개 이상의 TPU 코어입니다.