lớp cuối cùng công khai được phân vùng đầu ra
Một op phân tách một tensor được XLA phân chia thành một danh sách được phân vùng
đầu ra bên ngoài tính toán XLA.
Các lớp lồng nhau
| lớp học | Phân vùngOutput.Options | Các thuộc tính tùy chọn cho PartitionedOutput | |
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 <T mở rộng TType > Đầu ra được phân vùng <T> | tạo ( Phạm vi phạm vi, đầu vào Toán hạng <T>, Tùy chọn numSplits dài, 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 Phân vùng đầu ra mới. |
| Trình vòng lặp< Toán hạng <T>> | trình lặp () |
| Danh sách< Đầu ra <T>> | đầu ra () Danh sách các đầu vào được phân vùng phải có hình dạng giống nhau. |
| phân vùng tĩnh.Options | phân vùngDim (Phân vùng dàiDim) |
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: "TPUPartitionedOutput"
Phương pháp công khai
public static Phân vùng đầu ra <T> tạo (Phạm vi phạm vi , đầu vào Toán hạng <T>, numSplits 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 Phân vùng đầu ra mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| đầu vào | Một tensor đại diện cho hình dạng đầy đủ của các tensor được phân vùng. |
| 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 Phân vùng đầu ra
Danh sách công khai< Đầu ra <T>> đầu ra ()
Danh sách các đầu vào được phân vùng phải có hình dạng giống nhau.
công khai tĩnh Phân vùng đầu ra. Tùy chọn phân vùngDim (Phân vùng dàiDim)
Thông số
| phân vùng mờ | Một số nguyên mô tả kích thước nào được phân chia. |
|---|