ReplicateMetadata

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 allowSoftPlacement (Boolean allowSoftPlacement)

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 hostingComputeCore (Danh sách <String> hostingComputeCore)

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.

public static ReplicateMetadata.Options bản đồ đệm (Danh sách<String> bản đồ đệm)

tĩnh công khai ReplicateMetadata.Options stepMarkerLocation (Chuỗi stepMarkerLocation)

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.