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 |
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