lớp cuối cùng công khai ReplicateMetadata
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 | Sao chépMetadata.Options | Các thuộc tính tùy chọn cho ReplicateMetadata |
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
tĩnh ReplicateMetadata.Options | allowSoftPlacement (Boolean allowSoftPlacement) |
tĩnh ReplicateMetadata.Options | 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 sao chép 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 ReplicateMetadata mới. |
tĩnh ReplicateMetadata.Options | deviceAssignment (Danh sách<Long> deviceAssignment) |
tĩnh ReplicateMetadata.Options | máy chủComputeCore (Danh sách <Chuỗi> máy chủComputeCore) |
tĩnh ReplicateMetadata.Options | numCoresPerReplica (numCoresPerReplica dài) |
tĩnh ReplicateMetadata.Options | bản đồ đệm (Danh sách<String> bản đồ đệm) |
tĩnh ReplicateMetadata.Options | stepMarkerLocation (Chuỗi stepMarkerLocation) |
tĩnh ReplicateMetadata.Options | cấu trúc liên kết (cấu trúc liên kết chuỗi) |
tĩnh ReplicateMetadata.Options | useSpmdForXlaPartitioning (Sử dụng BooleanSpmdForXlaPartitioning) |
tĩnh ReplicateMetadata.Options | 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
tĩnh công khai ReplicateMetadata.Options tính toánShape (Danh sách 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 ReplicateMetadata 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 ReplicateMetadata 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 ReplicateMetadata
tĩnh công khai ReplicateMetadata.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 ReplicateMetadata.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 ReplicateMetadata.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. |
---|
public static ReplicateMetadata.Options useSpmdForXlaPartitioning (Boolean useSpmdForXlaPartitioning)
tĩnh công khai ReplicateMetadata.Options useTpu (Boolean useTpu)
Thông số
sử dụngTpu | Có nên đặt tính toán trên TPU hay không. |
---|