TPUReplicateMetadata.Options

공개 정적 클래스 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 허용SoftPlacement (부울 허용SoftPlacement)

공개 TPUReplicateMetadata.Options 계산Shape (List<Long> 계산Shape)

매개변수
계산모양 더 이상 사용되지 않습니다. 대신 num_cores_per_replica를 사용하세요.

공개 TPUReplicateMetadata.Options deviceAssignment (List<Long> deviceAssignment)

매개변수
장치할당 계산을 위한 장치 할당.

공개 TPUReplicateMetadata.Options 호스트ComputeCore (목록<문자열> 호스트ComputeCore)

공개 TPUReplicateMetadata.Options numCoresPerReplica (Long numCoresPerReplica)

매개변수
복제본당 코어 수 복제본당 코어 수입니다. 모델 병렬성에 사용됩니다.

공개 TPUReplicateMetadata.Options paddingMap (List<String> paddingMap)

공개 TPUReplicateMetadata.Options stepMarkerLocation (문자열 stepMarkerLocation)

공개 TPUReplicateMetadata.Options 토폴로지 (문자열 토폴로지)

매개변수
토폴로지 TPU Pod 슬라이스의 토폴로지를 나타내는 TopologyProto입니다.

공개 TPUReplicateMetadata.Options useSpmdForXlaPartitioning (부울 useSpmdForXlaPartitioning)

공개 TPUReplicateMetadata.Options useTpu (부울 useTpu)

매개변수
용도Tpu TPU에 계산을 배치할지 여부입니다.