공개 정적 클래스 TPUReplicateMetadata.Options
TPUReplicateMetadata
의 선택적 속성
공개 방법
TPUReplicateMetadata.Options | allowedSoftPlacement (부울 허용SoftPlacement) |
TPUReplicateMetadata.Options | 계산 셰이프 (List<Long> 계산 셰이프) |
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.Options deviceAssignment (List<Long> deviceAssignment)
매개변수
장치할당 | 계산을 위한 장치 할당. |
---|
공개 TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)
매개변수
복제본당 코어 수 | 복제본당 코어 수입니다. 모델 병렬성에 사용됩니다. |
---|
공개 TPUReplicateMetadata.Options 토폴로지 (문자열 토폴로지)
매개변수
토폴로지 | TPU Pod 슬라이스의 토폴로지를 나타내는 TopologyProto입니다. |
---|