공개 최종 클래스 TPURoundRobin
TPU 코어의 라운드 로빈 부하 분산.
TPU 코어 간에 라운드 로빈을 수행하는 부하 분산 작업입니다.
이 작업은 [0, NumTPUCoresVisiblePerHost]의 정수 사이에서 라운드 로빈됩니다. 'TPUPPartitionedCall'과 같이 계산을 실행할 TPU 코어를 입력으로 사용하는 TensorFlow 작업과 인터페이스하는 데 유용합니다.
device_ordinal: [0, NumTPUCoresVisiblePerHost]의 정수입니다.
공개 방법
출력 <정수> | 출력 () 텐서의 기호 핸들을 반환합니다. |
정적 TPURoundRobin | |
출력 <정수> | 장치서수 () |
상속된 메서드
공개 방법
공개 출력 <Integer> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
공개 정적 TPURoundRobin 생성 ( 범위 범위)
새 TPURoundRobin 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|
보고
- TPURoundRobin의 새 인스턴스