TPUReplicateMetadata

공개 최종 클래스 TPUReplicateMetadata

TPU 계산을 복제하는 방법을 나타내는 메타데이터입니다.

이 작업은 'tpu.replicate()' 계산 하위 그래프 작업에 공통적인 메타데이터를 보유합니다.

중첩 클래스

수업 TPUReplicateMetadata.Options TPUReplicateMetadata 의 선택적 속성

상수

OP_NAME TensorFlow 코어 엔진에서 알려진 이 작업의 이름

공개 방법

정적 TPUReplicateMetadata.Options
allowedSoftPlacement (부울 허용SoftPlacement)
정적 TPUReplicateMetadata.Options
계산 셰이프 (List<Long> 계산 셰이프)
정적 TPUReplicateMetadata
생성 ( 범위 범위, Long numReplicas, 옵션... 옵션)
새 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)

상속된 메서드

상수

공개 정적 최종 문자열 OP_NAME

TensorFlow 코어 엔진에서 알려진 이 작업의 이름

상수 값: "TPUReplicateMetadata"

공개 방법

공개 정적 TPUReplicateMetadata.Options 허용SoftPlacement (부울 허용SoftPlacement)

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

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

공개 정적 TPUReplicateMetadata 생성 ( 범위 범위, Long numReplicas, 옵션... 옵션)

새 TPUReplicateMetadata 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

매개변수
범위 현재 범위
수복제본 계산의 복제본 수
옵션 선택적 속성 값을 전달합니다.
보고
  • TPUReplicateMetadata의 새 인스턴스

공개 정적 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에 계산을 배치할지 여부입니다.