ExecuteTPUEmbeddingPartitioner

lớp cuối cùng công khai Thực thiTPUEmbeddingPartitioner

Một op thực thi trình phân vùng TPUEmbedding trên cấu hình trung tâm

thiết bị và tính toán kích thước HBM (tính bằng byte) cần thiết cho hoạt động TPUEmbedding.

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

Đầu ra <Chuỗi>
asOutput ()
Trả về phần điều khiển tượng trưng của một tenxơ.
Đầu ra <Chuỗi>
commonConfig ()
Một proto cấu hình chung được mã hóa chuỗi chứa siêu dữ liệu về đầu ra của trình phân vùng TPUEmbedding và kích thước HBM (tính bằng byte) cần thiết để hoạt động.
Thực thi tĩnhTPUEmbeddingPartitioner
tạo ( Phạm vi phạm vi, Cấu hình chuỗi)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác Thực thiTPUEmbeddingPartitioner mới.

Phương pháp kế thừa

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

Đầu ra công khai <String> asOutput ()

Trả về phần điều khiển tượng trưng của một tenxơ.

Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.

Đầu ra công khai <String> commonConfig ()

Một proto cấu hình chung được mã hóa chuỗi chứa siêu dữ liệu về đầu ra của trình phân vùng TPUEmbedding và kích thước HBM (tính bằng byte) cần thiết để hoạt động.

tạo công khai tĩnh Thực thiTPUEmbeddingPartitioner (Phạm vi phạm vi , Cấu hình chuỗi)

Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác Thực thiTPUEmbeddingPartitioner mới.

Thông số
phạm vi phạm vi hiện tại
cấu hình Một proto TPUEmbeddingConfiguration được tuần tự hóa thành một chuỗi, mô tả cấu hình TPUEmbedding mong muốn.
Trả lại
  • một phiên bản mới của Thực thiTPUEmbeddingPartitioner