공개 최종 클래스 TPUReplicateMetadata
TPU 계산을 복제하는 방법을 나타내는 메타데이터입니다.
이 작업은 'tpu.replicate()' 계산 하위 그래프 작업에 공통적인 메타데이터를 보유합니다.
중첩 클래스
| 수업 | TPUReplicateMetadata.Options | TPUReplicateMetadata 의 선택적 속성 | |
공개 방법
| 정적 TPUReplicateMetadata.Options | allowedSoftPlacement (부울 허용SoftPlacement) |
| 정적 TPUReplicateMetadata.Options | 계산 셰이프 (List<Long> 계산 셰이프) |
| 정적 TPUReplicateMetadata | |
| 정적 TPUReplicateMetadata.Options | deviceAssignment (List<Long> deviceAssignment) |
| 정적 TPUReplicateMetadata.Options | HostComputeCore (목록<문자열> 호스트ComputeCore) |
| 정적 TPUReplicateMetadata.Options | numCoresPerReplica (긴 numCoresPerReplica) |
| 정적 TPUReplicateMetadata.Options | paddingMap (List<String> paddingMap) |
| 정적 TPUReplicateMetadata.Options | stepMarkerLocation (문자열 stepMarkerLocation) |
| 정적 TPUReplicateMetadata.Options | 토폴로지 (문자열 토폴로지) |
| 정적 TPUReplicateMetadata.Options | useSpmdForXlaPartitioning (부울 useSpmdForXlaPartitioning) |
| 정적 TPUReplicateMetadata.Options | useTpu (부울 useTpu) |
상속된 메서드
공개 방법
공개 정적 TPUReplicateMetadata.Options 계산Shape (List<Long> 계산Shape)
매개변수
| 계산모양 | 더 이상 사용되지 않습니다. 대신 num_cores_per_replica를 사용하세요. |
|---|
공개 정적 TPUReplicateMetadata 생성 ( 범위 범위, Long numReplicas, 옵션... 옵션)
새 TPUReplicateMetadata 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
| 범위 | 현재 범위 |
|---|---|
| 수복제본 | 계산의 복제본 수 |
| 옵션 | 선택적 속성 값을 전달합니다. |
보고
- TPUReplicateMetadata의 새 인스턴스
공개 정적 TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)
매개변수
| 장치할당 | 계산을 위한 장치 할당. |
|---|
공개 정적 TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
매개변수
| 복제본당 코어 수 | 복제본당 코어 수입니다. 모델 병렬성에 사용됩니다. |
|---|
공개 정적 TPUReplicateMetadata.Options 토폴로지 (문자열 토폴로지)
매개변수
| 토폴로지 | TPU Pod 슬라이스의 토폴로지를 나타내는 TopologyProto입니다. |
|---|