lớp cuối cùng công khai TPUReplicateMetadata
Siêu dữ liệu cho biết cách tính toán TPU nên được sao chép.
Hoạt động này chứa siêu dữ liệu chung cho các hoạt động của biểu đồ con tính toán `tpu.replicate()`.
Các lớp lồng nhau
| lớp học | TPUReplicateMetadata.Options | Thuộc tính tùy chọn cho TPUReplicateMetadata | |
Hằng số
| Sợi dây | OP_NAME | Tên của op này, được biết đến bởi công cụ lõi TensorFlow |
Phương pháp công khai
| TPUReplicateMetadata.Options tĩnh | allowSoftPlacement (Boolean allowSoftPlacement) |
| TPUReplicateMetadata.Options tĩnh | hình dạng tính toán (Danh sách <Dài> hình dạng tính toán) |
| siêu dữ liệu TPUReplicateMetadata tĩnh | tạo ( Phạm vi phạm vi, Số bản sao dài, Tùy chọn... tùy chọn) Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TPUReplicateMetadata mới. |
| TPUReplicateMetadata.Options tĩnh | deviceAssignment (Danh sách<Long> deviceAssignment) |
| TPUReplicateMetadata.Options tĩnh | máy chủComputeCore (Danh sách <Chuỗi> máy chủComputeCore) |
| TPUReplicateMetadata.Options tĩnh | numCoresPerReplica (numCoresPerReplica dài) |
| TPUReplicateMetadata.Options tĩnh | bản đồ đệm (Danh sách<String> bản đồ đệm) |
| TPUReplicateMetadata.Options tĩnh | stepMarkerLocation (Chuỗi stepMarkerLocation) |
| TPUReplicateMetadata.Options tĩnh | cấu trúc liên kết (cấu trúc liên kết chuỗi) |
| TPUReplicateMetadata.Options tĩnh | useSpmdForXlaPartitioning (Sử dụng BooleanSpmdForXlaPartitioning) |
| TPUReplicateMetadata.Options tĩnh | useTpu (sử dụng BooleanTpu) |
Phương pháp kế thừa
Hằng số
Chuỗi cuối cùng tĩnh công khai OP_NAME
Tên của op này, được biết đến bởi công cụ lõi TensorFlow
Giá trị không đổi: "TPURlicateMetadata"
Phương pháp công khai
hình dạng tính toán TPUReplicateMetadata.Options tĩnh 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 tĩnh TPUReplicateMetadata tạo ( Phạm vi phạm vi , Số bản sao dài, Tùy chọn... tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TPUReplicateMetadata mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| sốBản sao | Số lượng bản sao của tính toán |
| tùy chọn | mang các giá trị thuộc tính tùy chọn |
Trả lại
- một phiên bản mới của TPUReplicateMetadata
công khai tĩnh 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. |
|---|
tĩnh công khai TPUReplicateMetadata.Options hostingComputeCore (Danh sách <String> hostingComputeCore)
tĩnh 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 tĩnh công khai TPUReplicateMetadata.Options (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. |
|---|
tĩnh công khai TPUReplicateMetadata.Options useSpmdForXlaPartitioning (Sử dụng BooleanSpmdForXlaPartitioning)
tĩnh 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. |
|---|