TPUReplicateMetadata.Options

lớp tĩnh công khai TPUReplicateMetadata.Options

Thuộc tính tùy chọn cho TPUReplicateMetadata

Phương pháp công khai

TPUReplicateMetadata.Options
allowSoftPlacement (Boolean allowSoftPlacement)
TPUReplicateMetadata.Options
hình dạng tính toán (Danh sách <Dài> hình dạng tính toán)
TPUReplicateMetadata.Options
deviceAssignment (Danh sách<Long> deviceAssignment)
TPUReplicateMetadata.Options
máy chủComputeCore (Danh sách <Chuỗi> máy chủComputeCore)
TPUReplicateMetadata.Options
numCoresPerReplica (numCoresPerReplica dài)
TPUReplicateMetadata.Options
bản đồ đệm (Danh sách<String> bản đồ đệm)
TPUReplicateMetadata.Options
stepMarkerLocation (Chuỗi stepMarkerLocation)
TPUReplicateMetadata.Options
cấu trúc liên kết (cấu trúc liên kết chuỗi)
TPUReplicateMetadata.Options
useSpmdForXlaPartitioning (Sử dụng BooleanSpmdForXlaPartitioning)
TPUReplicateMetadata.Options
useTpu (sử dụng BooleanTpu)

Phương pháp kế thừa

Phương pháp công khai

công khai TPUReplicateMetadata.Options allowSoftPlacement (Boolean allowSoftPlacement)

hình dạng tính toán TPUReplicateMetadata.Options công khai (Danh sách hình dạng tính toán <Dài>)

Thông số
hình dạng tính toán KHÔNG ĐƯỢC DÙNG. Thay vào đó hãy sử dụng num_cores_per_replica.

công khai TPUReplicateMetadata.Options deviceAssignment (Danh sách<Long> deviceAssignment)

Thông số
bài tập thiết bị Việc phân công các thiết bị để tính toán.

công khai TPUReplicateMetadata.Options HostComputeCore (Danh sách<String> HostComputeCore)

công khai TPUReplicateMetadata.Options numCoresPerReplica (numCoresPerReplica dài)

Thông số
numCoresPerBản sao Số lượng lõi trên mỗi bản sao. Được sử dụng cho mô hình song song.

bản đồ đệm TPUReplicateMetadata.Options công khai (Danh sách <String> bản đồ đệm)

công khai TPUReplicateMetadata.Options stepMarkerLocation (Chuỗi stepMarkerLocation)

cấu trúc liên kết TPUReplicateMetadata.Options công khai (Cấu trúc liên kết chuỗi)

Thông số
cấu trúc liên kết TopologyProto biểu thị cấu trúc liên kết của lát vỏ TPU.

public TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Sử dụng BooleanSpmdForXlaPartitioning)

công khai TPUReplicateMetadata.Options useTpu (Boolean useTpu)

Thông số
sử dụngTpu Có nên đặt tính toán trên TPU hay không.