이 페이지는 Cloud Translation API를 통해 번역되었습니다.
Switch to English

tf.tpu.experimental.DeviceAssignment

TensorFlow 1 개 버전 GitHub의에서 소스보기

TPU 물리적 토폴로지 계산 논리 코어로 매핑.

사용하는 것을 선호 DeviceAssignment.build() 구축 도우미를 DeviceAssignment ; 그것은 건설보다 쉽습니다 경우 덜 유연 DeviceAssignment 직접.

topology Topology 물리적 TPU 토폴로지를 설명 객체입니다.
core_assignment 코어 물리적 맵핑에 대한 논리, 랭크 3 NumPy와 배열로 표현. 의 설명을 참조하십시오 core_assignment 자세한 내용은 속성을.

ValueError 경우 topology 아니다 Topology 객체입니다.
ValueError 경우 core_assignment 순위 3 NumPy와 배열이 아닙니다.

core_assignment 물리적 코어 매핑 논리.
num_cores_per_replica 복제 당 코어 수입니다.
num_replicas 계산의 복제본 수.
topology Topology TPU의 토폴로지에 대해 설명합니다.

행동 양식

build

소스보기

coordinates

소스보기

논리적 코어의 물리적 토폴로지 좌표를 돌려줍니다.

host_device

소스보기

논리적 코어에 부착 된 CPU 장치를 돌려줍니다.

lookup_replicas

소스보기

작업 번호와 논리의 핵심으로 조회 복제 식별자.

인수
task_id TensorFlow 작업 번호.
logical_core 논리 코어를 식별하는 정수.

보고
A는 해당 작업과 logical_core에 부착 된 복제본의 목록을 분류.

제기
ValueError 복제본은 논리 코어를 포함하는 작업에 존재하지 않는 경우.

tpu_device

소스보기

논리적 코어에 할당 된 TPU 장치의 이름을 돌려줍니다.

tpu_ordinal

소스보기

논리적 코어에 할당 된 TPU 장치의 순서를 돌려줍니다.