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 | tpuCompileOptionsProto (Chuỗi tpuCompileOptionsProto) |
| 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
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 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. |
|---|
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. |
|---|