TPUReplicateMetadata

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

Phương pháp công cộng

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

Phương pháp công cộng

tĩnh công khai TPUReplicateMetadata.Options allowSoftPlacement (Boolean allowSoftPlacement)

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.

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

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

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

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.